diff --git a/pyLIMA/outputs/pyLIMA_plots.py b/pyLIMA/outputs/pyLIMA_plots.py
index b6ec7d8..2657469 100644
--- a/pyLIMA/outputs/pyLIMA_plots.py
+++ b/pyLIMA/outputs/pyLIMA_plots.py
@@ -127,8 +127,8 @@ def create_telescopes_to_plot_model(microlensing_model, pyLIMA_parameters):
 
                     model_time = np.r_[model_time1, model_time2,tel.lightcurve_magnitude['time'].value]
 
-                    mask = (model_time>tel.lightcurve_magnitude['time'].value.min()) &
-                    (model_time < tel.lightcurve_magnitude['time'].value.max())
+                    mask = (model_time > tel.astrometry['time'].value.min()) & (
+                            model_time < tel.lightcurve_astrometry['time'].value.max())
                     model_time = model_time[mask]
 
                     model_time.sort()
@@ -181,8 +181,8 @@ def create_telescopes_to_plot_model(microlensing_model, pyLIMA_parameters):
 
                 if tel.location == 'Space':
 
-                    mask = (model_time > tel.astrometry['time'].value.min()) &
-                    (model_time < tel.lightcurve_astrometry['time'].value.max())
+                    mask = (model_time > tel.astrometry['time'].value.min()) & (
+                            model_time < tel.lightcurve_astrometry['time'].value.max())
                     model_time = model_time[mask]
 
                 model_time.sort()