Fixes model code highlighting