Fixing minor bug in icon display