Fixed import/from/assert/as highlighting