Fixed another minor bug