@@ -318,7 +318,7 @@ class TestConfigCommands(unittest.TestCase):
self.tool.precmd(line)
self.tool._update_all_modules_info = update_all_modules_info
# If line is equals to 'EOF', _update_all_modules_info() shouldn't be called
- self.assertRaises(None, precmd('EOF'))
+ precmd('EOF')
self.assertRaises(socket.error, precmd, 'continue')
def test_run(self):