Fix height resize bug