diff --git a/comfy_extras/nodes_dataset.py b/comfy_extras/nodes_dataset.py
index 7df2548ec..4789d7d53 100644
--- a/comfy_extras/nodes_dataset.py
+++ b/comfy_extras/nodes_dataset.py
@@ -1,6 +1,5 @@
 import logging
 import os
-import math
 import json
 
 import numpy as np