Fix for reloadData to reflect the changes