fix issue for removing cells