WK-1086 fix tests