diff --git a/deepof/pose_utils.py b/deepof/pose_utils.py
index 08858ff87811ccf56ddb4c7a5ac061e96c581afa..5deb463bafa961642acd25dca9a08ad26a0bd515 100644
--- a/deepof/pose_utils.py
+++ b/deepof/pose_utils.py
@@ -19,6 +19,7 @@ import os
 import pandas as pd
 import regex as re
 import seaborn as sns
+import tensorflow as tf
 import warnings
 
 # Ignore warning with no downstream effect
@@ -1051,7 +1052,7 @@ def rule_based_video(
         recog_limit,
         coordinates._arena,
         detection_mode=coordinates._arena_detection,
-        cnn_model=self._ellipse_detection_model,
+        cnn_model=coordinates._ellipse_detection_model,
     )
     corners = frame_corners(h, w)