fix assert that block multiple distillation