WK-557

Fix the test