Correcting argument to callback on 'register'