diff --git a/skais/ais/ais_trajectory.py b/skais/ais/ais_trajectory.py
index 336f6da6d695870ec2f2af9dddd203042fde6d1c..dca3bcc230e78ae872fbcee55bcccc8709a54bdf 100644
--- a/skais/ais/ais_trajectory.py
+++ b/skais/ais/ais_trajectory.py
@@ -358,11 +358,11 @@ class AISTrajectory(AISPoints):
         elif type(features) is list:
             features_vectors = self.df[features].to_numpy()
             for c in features_vectors.T:
-                bounds.append((0, max(c)))
+                bounds.append([0, max(c)])
         elif type(features) is str:
             features_vectors = self.df[[features]].to_numpy()
             for c in features_vectors.T:
-                bounds.append((0, max(c)))
+                bounds.append([0, max(c)])
         elif type(features) is dict:
             bounds = list(features.values())
             features_vectors = self.df[features.keys()].to_numpy()
@@ -383,6 +383,6 @@ class AISTrajectory(AISPoints):
                 generate_links(data, positions, height, width, lower_lat, upper_lat, lower_lon, upper_lon)
 
         else:
-            generate_points_with_features(data, positions, features_vectors, bounds, node_size, height, width,
+            generate_points_with_features(data, positions, features_vectors, np.array(bounds), node_size, height, width,
                                           lower_lat, upper_lat, lower_lon, upper_lon, link)
         return data