diff --git a/example/50_put_element/put_element_fp16.cpp b/example/50_put_element/put_element_fp16.cpp
index d4b6831bc4..791747d875 100644
--- a/example/50_put_element/put_element_fp16.cpp
+++ b/example/50_put_element/put_element_fp16.cpp
@@ -34,9 +34,9 @@ int main()
 
     int N = 1024;
 
-    Tensor<XDataType> x(HostTensorDescriptor{N, 1});
-    Tensor<IndexDataType> indices(HostTensorDescriptor{N, 1});
-    Tensor<YDataType> y(HostTensorDescriptor{N, 1});
+    Tensor<XDataType> x(HostTensorDescriptor{N});
+    Tensor<IndexDataType> indices(HostTensorDescriptor{N});
+    Tensor<YDataType> y(HostTensorDescriptor{N});
 
     x.GenerateTensorValue(GeneratorTensor_3<XDataType>{-1.0, 1.0});
     for(int i = 0; i < N; ++i)
@@ -72,7 +72,7 @@ int main()
     bool pass = true;
     if(do_verification)
     {
-        Tensor<YDataType> y_host(HostTensorDescriptor{N, 1});
+        Tensor<YDataType> y_host(HostTensorDescriptor{N});
 
         for(int i = 0; i < N; ++i)
         {