Fixing #48

Forgot I changed the method signature.