Try to fix error in tests (2)