diff --git a/armory/data/datasets.py b/armory/data/datasets.py
index cec605402..5f6d96c26 100644
--- a/armory/data/datasets.py
+++ b/armory/data/datasets.py
@@ -828,7 +828,7 @@ def __init__(self, x_shape, sample_rate, input_type=np.int64):
 
 
 def canonical_audio_preprocess(context, batch):
-    if batch.dtype == np.object:
+    if batch.dtype == object:
         for x in batch:
             check_shapes(x.shape, context.x_shape)
             assert x.dtype == context.input_type
@@ -1692,8 +1692,8 @@ def __init__(self, max_frames):
         self.max_frames = max_frames
 
     def __call__(self, batch):
-        if batch.dtype == np.object:
-            clipped_batch = np.empty_like(batch, dtype=np.object)
+        if batch.dtype == object:
+            clipped_batch = np.empty_like(batch, dtype=object)
             clipped_batch[:] = [x[: self.max_frames] for x in batch]
             return clipped_batch
         else: