fixed another bug in tests