fixed a minor bug in their testing code