Fixed yet another minor bug