Add code for removing migration states when relevant spaces are removed