diff --git a/docs/DATASET.md b/docs/DATASET.md
index 1ae8bac8b2a3b20f5be8ea6c9f1f0abd70453245..9339c3a3a6c282990a3ba2fd03efe2594674e01c 100644
--- a/docs/DATASET.md
+++ b/docs/DATASET.md
@@ -143,12 +143,13 @@ NOTE: You don't need to have `groundtruth.json` pre-downloaded, it will automati
 flight_ids = dataset.get_flight_ids()
 flight = dataset.get_flight(flight_ids[0]) # Loading single flight
 
-airborne_objects = flight.get_airborne_objects() # Get all airborne objects
-airborne_objects.location # Location of object in whole flight
+airborne_objects = list(flight.get_airborne_objects()) # Get all airborne objects
+airborne_objects[0].location # Location of object in whole flight
 
 frames = flight.frames # All the frames of the flight
-airborne_objects_in_frame = frames.detected_objects
-airborne_objects_location_in_frame = frames.detected_object_locations
+frame_ids = list(frames.keys()) 
+airborne_objects_in_frame = frames[frame_ids[0]].detected_objects
+airborne_objects_location_in_frame = rames[frame_ids[0]].detected_object_locations
 
 # Check out the dataset exploration notebook for more...
 [...]