fix mock setup