Fix slight visual bug