update mock code