diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index 8c5e277..ab63afd 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -434,7 +434,7 @@
 
     def element_size(self) -> int:
         if self.element_size_bytes == 0:
-            return self.dtype.size_in_bits() / 8
+            return self.dtype.size_in_bits() // 8
         return self.element_size_bytes
 
     # Returns a copy, renamed to self.name + suffix