Fix the height problem