diff --git a/miplearn/branching.py b/miplearn/branching.py
index 40c2561..0654953 100644
--- a/miplearn/branching.py
+++ b/miplearn/branching.py
@@ -42,7 +42,8 @@ class BranchPriorityComponent(Component):
                            check=True,
                            capture_output=True)
             self._merge(np.genfromtxt(priority_file.name,
-                                      delimiter=','))
+                                      delimiter=',',
+                                      dtype=np.float64))
         
     
     def fit(self, solver):
diff --git a/miplearn/tests/test_solver.py b/miplearn/tests/test_solver.py
index 2e97fab..fc7c532 100644
--- a/miplearn/tests/test_solver.py
+++ b/miplearn/tests/test_solver.py
@@ -56,7 +56,7 @@ def test_parallel_solve():
 def test_solver_random_branch_priority():
     instance = _get_instance()
     components = {
-        "warm-start": BranchPriorityComponent(initial_priority=np.array([1, 2, 3, 4])),
+        "warm-start": BranchPriorityComponent(initial_priority=np.array([1., 2., 3., 4.])),
     }
     solver = LearningSolver(components=components)
     solver.solve(instance)