small fix function highlighting