diff --git a/AGV_quantum/qubo_solver.py b/AGV_quantum/qubo_solver.py
index 7800f3d..a72226a 100644
--- a/AGV_quantum/qubo_solver.py
+++ b/AGV_quantum/qubo_solver.py
@@ -1,4 +1,4 @@
-from typing import Tuple
+from typing import Dict, Tuple
 
 import os
 import neal