Fix method not found exception

:dancers: :coffee: :dancers: