A minor adjustment in tests