Fix issue where number is float now and needs to be int