Try to fix error in tests