diff --git a/run_detector.py b/run_detector.py
index 0fcb74e..0405e01 100644
--- a/run_detector.py
+++ b/run_detector.py
@@ -39,7 +39,7 @@ def draw_bboxes(image, detections):
     return image
 
 
-def run_detector(model, args):
+def run_detector(model: footandball.FootAndBall, args: argparse.Namespace):
     model.print_summary(show_architecture=False)
     model = model.to(args.device)