Fix height issue again