small fix for statement highlight