Fix container height issue