Fix create_mock_connection received unexpected argument bug