remove mock code