diff --git a/rocketpy/environment/environment.py b/rocketpy/environment/environment.py
index 7a215b022..dbabeee78 100644
--- a/rocketpy/environment/environment.py
+++ b/rocketpy/environment/environment.py
@@ -688,7 +688,9 @@ def set_date(self, date, timezone="UTC"):
             "Ensemble",
         ]:
             self.set_atmospheric_model(
-                type=self.atmospheric_model_type, file=self.atmospheric_model_file, dictionary=self.atmospheric_model_dict
+                type=self.atmospheric_model_type,
+                file=self.atmospheric_model_file,
+                dictionary=self.atmospheric_model_dict,
             )
 
     def set_location(self, latitude, longitude):
@@ -726,7 +728,9 @@ def set_location(self, latitude, longitude):
             "Ensemble",
         ]:
             self.set_atmospheric_model(
-                type=self.atmospheric_model_type, file=self.atmospheric_model_file, dictionary=self.atmospheric_model_dict
+                type=self.atmospheric_model_type,
+                file=self.atmospheric_model_file,
+                dictionary=self.atmospheric_model_dict,
             )
 
     def set_gravity_model(self, gravity=None):