Replace behaviour_info callback with callback attributes