Clean up on_* handlers per PR feedback