Fix missing highlight on print keyword