Fixing Diagnosis code not displaying bug