Fix a minor bug annoying bug