Corrected small bug in previous commits