Fixing small visual bug