Fix highlighting of mut