fix a small bug introduced into test