Code display fix