Fixed a bug in heightForRow