diff --git a/TinySTL/Test/SuffixArrayTest.cpp b/TinySTL/Test/SuffixArrayTest.cpp
index 658e17f..fa1bc59 100644
--- a/TinySTL/Test/SuffixArrayTest.cpp
+++ b/TinySTL/Test/SuffixArrayTest.cpp
@@ -3,9 +3,11 @@
 namespace TinySTL{
 	namespace SuffixArrayTest{
 		void testCase(){
-			char arr[] = { 'a', 'a', 'b', 'a', 'a', 'a', 'a', 'b' };
+			//char arr[] = { 'a', 'a', 'b', 'a', 'a', 'a', 'a', 'b' };
+			std::string str("aabaaaab");
 
-			TinySTL::suffix_array sa(arr, 8);
+			//TinySTL::suffix_array sa(arr, 8);
+			TinySTL::suffix_array sa(str.data(), str.size());
 			auto sa1 = sa.suffixArray();
 			auto sa2 = TinySTL::suffix_array::array_type{3, 4, 5, 0, 6, 1, 7, 2};
 			assert(TinySTL::Test::container_equal(sa1, sa2));
diff --git a/TinySTL/Test/SuffixArrayTest.h b/TinySTL/Test/SuffixArrayTest.h
index e6ff62b..a0c801b 100644
--- a/TinySTL/Test/SuffixArrayTest.h
+++ b/TinySTL/Test/SuffixArrayTest.h
@@ -6,6 +6,7 @@
 #include "../SuffixArray.h"
 
 #include <cassert>
+#include <string>
 
 namespace TinySTL{
 	namespace SuffixArrayTest{