diff --git a/dataloaders/am_dataloader.py b/dataloaders/am_dataloader.py
index 41f2dca..ea64693 100644
--- a/dataloaders/am_dataloader.py
+++ b/dataloaders/am_dataloader.py
@@ -39,7 +39,7 @@ def return_data_types(self):
         if self.LAS:
             return (tf.float32, tf.float32, tf.int32, tf.int32, tf.int32,tf.float32)
         else:
-            return  (tf.float32, tf.float32, tf.int32, tf.int32, tf.int32)
+            return  (tf.float32, tf.int32, tf.int32, tf.int32)
     def return_data_shape(self):
         f,c=self.speech_featurizer.compute_feature_dim()
         if self.LAS: