Fixing method names in tests