Quick fix

fixed a small bug