Add removal of observer in test