diff --git a/pyLIMA/outputs/pyLIMA_plots.py b/pyLIMA/outputs/pyLIMA_plots.py
index 2657469..d21e1dd 100644
--- a/pyLIMA/outputs/pyLIMA_plots.py
+++ b/pyLIMA/outputs/pyLIMA_plots.py
@@ -127,8 +127,9 @@ 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.astrometry['time'].value.min()) & (
-                            model_time < tel.lightcurve_astrometry['time'].value.max())
+                    mask = ((model_time > tel.lightcurve_magnitude['time'].value.min())
+                            & (
+                            model_time < tel.lightcurve_magnitude['time'].value.max()))
                     model_time = model_time[mask]
 
                     model_time.sort()
@@ -182,7 +183,7 @@ 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())
+                            model_time < tel.astrometry['time'].value.max())
                     model_time = model_time[mask]
 
                 model_time.sort()