MLDS-531 Test fix