Fix a small bug in the scheme