Fix tests for method call