fix epoch choose bug