Fixed the height issue for mobile