diff --git a/inference/rules/algorithms.py b/inference/rules/algorithms.py
index d2fc4b9..78a462b 100644
--- a/inference/rules/algorithms.py
+++ b/inference/rules/algorithms.py
@@ -9,6 +9,8 @@ import numpy as np
 
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
+from sort import Sort
+
 
 class LeavePostAlgorithm:
     STATE_INIT = "INIT_STATE"