Fixed visual bug