Fixed the height issue