fixed a small bug in previous commit