fix visual bug