diff --git a/unittest/tensor.cpp b/unittest/tensor.cpp
index c770112b3f26c9e65c960f8faf9e9b65c10057e6..80838ebfc06d0a810d4d252f3c9322220a4850bb 100644
--- a/unittest/tensor.cpp
+++ b/unittest/tensor.cpp
@@ -122,7 +122,7 @@ template <typename Scalar, int Rank>
 struct TensorContainer {
   typedef Eigen::Tensor<Scalar, Rank> Tensor;
   typedef Eigen::TensorRef<Tensor> TensorRef;
-  typedef Eigen::Matrix<int, Rank, 1> Dimensions;
+  typedef Eigen::Matrix<Scalar, Rank, 1> Dimensions;
 
   Tensor m_tensor;
   TensorContainer(const Dimensions& dims) {