Fix height issue