Fix the last visual bug