diff --git a/STDStream.cpp b/STDStream.cpp
index 762c5d9..cd966ae 100644
--- a/STDStream.cpp
+++ b/STDStream.cpp
@@ -15,7 +15,7 @@
 auto exe_policy = std::execution::par_unseq;
 
 template <class T>
-STDStream<T>::STDStream(const unsigned int ARRAY_SIZE, int device)
+STDStream<T>::STDStream(const int ARRAY_SIZE, int device)
   noexcept : array_size{ARRAY_SIZE}, a{new T[array_size]}, b{new T[array_size]}, c{new T[array_size]}
 {
 }
diff --git a/STDStream.h b/STDStream.h
index f24165f..538f857 100644
--- a/STDStream.h
+++ b/STDStream.h
@@ -16,7 +16,7 @@ class STDStream : public Stream<T>
 {
   protected:
     // Size of arrays
-    unsigned int array_size;
+    int array_size;
 
     // Device side pointers
     T *a;
@@ -24,7 +24,7 @@ class STDStream : public Stream<T>
     T *c;
 
   public:
-    STDStream(const unsigned int, int);
+    STDStream(const int, int);
     ~STDStream();
 
     virtual void copy() override;