fix mock build