KAA-234: Test fix