fix epochMap to work as an example