Fixed the implementation for ReplayController