Update @on_definition docs