Add tests around hook behaviour
Fix failing test to preserve behaviour