Fix the story removing bug