diff --git a/OCLStream.cpp b/OCLStream.cpp
index 4d6e85a..7e1bcf0 100644
--- a/OCLStream.cpp
+++ b/OCLStream.cpp
@@ -92,7 +92,7 @@ std::string kernels{R"CLC(
 
 
 template <class T>
-OCLStream<T>::OCLStream(const unsigned int ARRAY_SIZE, const int device_index)
+OCLStream<T>::OCLStream(const int ARRAY_SIZE, const int device_index)
 {
   if (!cached)
     getDeviceList();
diff --git a/OCLStream.h b/OCLStream.h
index 02ffe44..3085aca 100644
--- a/OCLStream.h
+++ b/OCLStream.h
@@ -26,7 +26,7 @@ class OCLStream : public Stream<T>
 {
   protected:
     // Size of arrays
-    unsigned int array_size;
+    int array_size;
 
     // Host array for partial sums for dot kernel
     std::vector<T> sums;
@@ -55,7 +55,7 @@ class OCLStream : public Stream<T>
 
   public:
 
-    OCLStream(const unsigned int, const int);
+    OCLStream(const int, const int);
     ~OCLStream();
 
     virtual void copy() override;