OZP-770: fix tests