Fixed small bug in previous commit