diff --git a/vitookit/evaluation/eval_cls_ffcv.py b/vitookit/evaluation/eval_cls_ffcv.py
index 2450174e72c8c17c59ebef3bb29cabc9fa6bb9d7..c737aa10d4f5c48cd780deec062051dbc911a12c 100644
--- a/vitookit/evaluation/eval_cls_ffcv.py
+++ b/vitookit/evaluation/eval_cls_ffcv.py
@@ -27,6 +27,7 @@ from vitookit.evaluation.eval_cls import evaluate
 from vitookit.utils.helper import *
 from vitookit.utils import misc
 from vitookit.models.build_model import build_model
+from vitookit.datasets.dres import DynamicResolution
 import wandb
 
 from pathlib import Path
@@ -43,7 +44,6 @@ from timm.layers import trunc_normal_
 
 from ffcv import Loader
 from ffcv.loader import OrderOption
-from datasets.dres import DynamicResolution
 
 def get_args_parser():
     parser = argparse.ArgumentParser('DeiT training and evaluation script', add_help=False)