Correct visual bug