Fix customized height issue