Fixed an minor bug