diff --git a/src/tensor.h b/src/tensor.h
index 66690427..40bca4f8 100644
--- a/src/tensor.h
+++ b/src/tensor.h
@@ -178,12 +178,12 @@ class Tensor {
       return pimpl_->begin();
     }
 
-    auto end() -> decltype( pimpl_->begin() ) {
-      return pimpl_->begin();
+    auto end() -> decltype( pimpl_->end() ) {
+      return pimpl_->end();
     }
 
-    auto end() const -> decltype( pimpl_->begin() ) {
-      return pimpl_->begin();
+    auto end() const -> decltype( pimpl_->end() ) {
+      return pimpl_->end();
     }
 
     const Shape& shape() const {