DB-397 fix tests