diff --git a/+Simulator/@TwoDimensionalMOT/runSimulation.m b/+Simulator/@TwoDimensionalMOT/runSimulation.m
index 36fe7c4..f26b3b1 100644
--- a/+Simulator/@TwoDimensionalMOT/runSimulation.m
+++ b/+Simulator/@TwoDimensionalMOT/runSimulation.m
@@ -1,11 +1,12 @@
 function [LoadingRate, StandardError, ConfidenceInterval] = runSimulation(this, ovenObj)
-    
+    if this.NumberOfAtoms ~= ovenObj.NumberOfAtoms
+        ovenObj.NumberOfAtoms = this.NumberOfAtoms;
+    end
     %% - Sampling for initial positions and velocities
     % - sampling the position distribution
     Positions  = ovenObj.initialPositionSampling();
     % - sampling the velocity distribution
     Velocities = ovenObj.initialVelocitySampling(this);
-    
     %% Solve ODE 
     progressbar  = Helper.parforNotifications();
     progressbar.PB_start(this.NumberOfAtoms,'Message',['Simulating 2-D MOT capture process for ' num2str(this.NumberOfAtoms,'%.0f') ' atoms:']);