diff --git a/KokkosStream.cpp b/KokkosStream.cpp
index 19d4b24..09c1eaf 100644
--- a/KokkosStream.cpp
+++ b/KokkosStream.cpp
@@ -9,7 +9,7 @@
 
 template <class T>
 KokkosStream<T>::KokkosStream(
-        const unsigned int ARRAY_SIZE, const int device_index)
+        const int ARRAY_SIZE, const int device_index)
     : array_size(ARRAY_SIZE)
 {
   Kokkos::initialize();
diff --git a/KokkosStream.hpp b/KokkosStream.hpp
index 1470cf4..e88622c 100644
--- a/KokkosStream.hpp
+++ b/KokkosStream.hpp
@@ -22,7 +22,7 @@ class KokkosStream : public Stream<T>
 {
   protected:
     // Size of arrays
-    unsigned int array_size;
+    int array_size;
 
     // Device side pointers to arrays
      typename Kokkos::View<T*>* d_a;
@@ -34,7 +34,7 @@ class KokkosStream : public Stream<T>
 
   public:
 
-    KokkosStream(const unsigned int, const int);
+    KokkosStream(const int, const int);
     ~KokkosStream();
 
     virtual void copy() override;