MDL-38538 Update unit tests