fix mock test