Fix Behaviour.getCallback