BS-11477
Fix unit test