diff --git a/tests/observation/test_multi_station_observation.py b/tests/observation/test_multi_station_observation.py
index b01cbf14d4d13830b7424030695307caab35a4d7..619918632ebe16676f6eaf77bad609d4b30d0653 100644
--- a/tests/observation/test_multi_station_observation.py
+++ b/tests/observation/test_multi_station_observation.py
@@ -6,23 +6,21 @@
 # See LICENSE.txt for more info.
 
 from unittest import mock
-
-
-from tests import base
+import concurrent.futures
 from json import loads
+from tests import base
 
 from lofar_station_client.observation.multi_station_observation import (
     MultiStationObservation,
 )
 
 
-import concurrent.futures
-
 SPEC_DICT = loads(
     """
         {
           "observation_id": 12345,
           "stop_time": "2106-02-07T00:00:00",
+          "antenna_set": "ALL",
           "antenna_mask": [0,1,2,9],
           "filter": "HBA_110_190",
           "SAPs": [{