diff --git a/ethosu/regor/compiler/scheduler_packing.cpp b/ethosu/regor/compiler/scheduler_packing.cpp
index 8d341976b2f1272c1f4784ee821c8def31da79d2..daae6f9e0490933b0b5ee1bd5c39f0b087fc58ab 100644
--- a/ethosu/regor/compiler/scheduler_packing.cpp
+++ b/ethosu/regor/compiler/scheduler_packing.cpp
@@ -284,6 +284,7 @@ void SchedulerPacking::HandleReinterpretCast(const Operation *op)
     if ( _graph->IsOutput(ofmConn->tensor.get()) )
     {
         InitSchedulerTensor(schedTensor.get(), ofmConn->tensor.get());
+        schedTensor->dataType = ifmConn->tensor->Type();
         schedTensor->srcTensor = ofmConn->tensor;
     }
 }