diff --git a/nova/tests/scheduler/test_least_cost.py b/nova/tests/scheduler/test_least_cost.py
index 64cda0b2..e64cc9e8 100644
--- a/nova/tests/scheduler/test_least_cost.py
+++ b/nova/tests/scheduler/test_least_cost.py
@@ -40,8 +40,7 @@ class LeastCostTestCase(test.TestCase):
         ctxt = context.get_admin_context()
         fakes.mox_host_manager_db_calls(self.mox, ctxt)
         self.mox.ReplayAll()
-        host_states = self.host_manager.get_all_host_states(ctxt,
-                'compute')
+        host_states = self.host_manager.get_all_host_states(ctxt)
         self.mox.VerifyAll()
         self.mox.ResetAll()
         return host_states
@@ -96,7 +95,7 @@ class TestWeightedHost(test.TestCase):
         self.assertDictMatch(host.to_dict(), expected)
 
     def test_dict_conversion_with_host_state(self):
-        host_state = host_manager.HostState('somehost', 'sometopic')
+        host_state = host_manager.HostState('somehost')
         host = least_cost.WeightedHost('someweight', host_state)
         expected = {'weight': 'someweight',
                     'host': 'somehost'}