fix mock after changes