Fix visual bug
Fix bug