diff --git a/neat/locals/overload/mhod/bruteforce.py b/neat/locals/overload/mhod/bruteforce.py
index 9231511..8b2c060 100644
--- a/neat/locals/overload/mhod/bruteforce.py
+++ b/neat/locals/overload/mhod/bruteforce.py
@@ -73,8 +73,8 @@ def optimize(step, limit, otf, migration_time, ls, p, state_vector,
     :param otf: The OTF parameter.
      :type otf: number,>=0,<=1
 
-    :param migration_time: The VM migration time in seconds.
-     :type migration_time: int,>=0
+    :param migration_time: The VM migration time in time steps.
+     :type migration_time: float,>=0
 
     :param ls: L functions.
      :type ls: list(function)
diff --git a/tests/locals/overload/mhod/test_bruteforce.py b/tests/locals/overload/mhod/test_bruteforce.py
index b07124e..17af9e0 100644
--- a/tests/locals/overload/mhod/test_bruteforce.py
+++ b/tests/locals/overload/mhod/test_bruteforce.py
@@ -60,7 +60,7 @@ class Bruteforce(TestCase):
             step = 0.1
             limit = 1
             otf = 0.3
-            migration_time = 20
+            migration_time = 20.
             ls = [lambda x: x, lambda x: x]
             p = [[0, 1]]
             state_vector = [0, 1]