"Fix bug in tests"