fixed minor bug in tests