Fixed another minor issue