diff --git a/deepfake.py b/deepfake.py
index 023b238..0637ce8 100644
--- a/deepfake.py
+++ b/deepfake.py
@@ -6,10 +6,10 @@ import time
 
 import torch
 import pandas as pd
-from kernel_utils import VideoReader, FaceExtractor, confident_strategy, predict_on_video_set
-from classifiers import DeepFakeClassifier
+from .kernel_utils import VideoReader, FaceExtractor, confident_strategy, predict_on_video_set
+from .classifiers import DeepFakeClassifier
 import towhee
-from towhee.operator.base import Operator, OperatorFlag
+from towhee.operator.base import NNOperator, OperatorFlag
 from towhee import register
 import warnings
 warnings.filterwarnings('ignore')
@@ -19,7 +19,7 @@ log = logging.getLogger()
 @register(output_schema=["scorelist"],
           flag=OperatorFlag.STATELESS | OperatorFlag.REUSEABLE)
 
-class deepfake(Operator):
+class Deepfake(NNOperator):
     '''
     ViT Distillation
     '''