Fix visual bug