diff --git a/src/lte/test/test-lte-epc-e2e-data.cc b/src/lte/test/test-lte-epc-e2e-data.cc
index f5045d777..14ee96a2b 100644
--- a/src/lte/test/test-lte-epc-e2e-data.cc
+++ b/src/lte/test/test-lte-epc-e2e-data.cc
@@ -272,16 +272,16 @@ LteEpcE2eDataTestCase::DoRun ()
                UintegerValue (2 * 1024 * 1024));
 
 
-  Time simulationTime = Seconds (2.0);
-
-  double statsStartTime = 0.040; // need to allow for RRC connection establishment + SRS 
+  double statsStartTime = 0.040; // need to allow for RRC connection establishment + SRS
+  double statsDuration = 2.0;
+    
   lteHelper->EnablePdcpTraces ();
 
   lteHelper->GetPdcpStats ()->SetAttribute ("StartTime", TimeValue (Seconds (statsStartTime)));
-  lteHelper->GetPdcpStats ()->SetAttribute ("EpochDuration", TimeValue (simulationTime));
+  lteHelper->GetPdcpStats ()->SetAttribute ("EpochDuration", TimeValue (Seconds (statsDuration)));
   
   
-  Simulator::Stop (simulationTime);  
+  Simulator::Stop (Seconds (statsStartTime + statsDuration - 0.0001));  
   Simulator::Run ();
 
   uint64_t imsiCounter = 0;