Fixed one more small bug