diff --git a/src/main/java/org/orekit/rugged/utils/DSGenerator.java b/src/main/java/org/orekit/rugged/utils/DSGenerator.java
index d757e2ea6fcb5a873f88a1eb549f41b2e5cf94ac..9c9d87b5076722d30e58a08407a9d02310cc5d7d 100644
--- a/src/main/java/org/orekit/rugged/utils/DSGenerator.java
+++ b/src/main/java/org/orekit/rugged/utils/DSGenerator.java
@@ -20,6 +20,7 @@ import java.util.List;
 
 import org.hipparchus.analysis.differentiation.DerivativeStructure;
 import org.orekit.utils.ParameterDriver;
+import org.orekit.utils.ParameterDriversList;
 
 /** Generator for {@link DerivativeStructure} instances from {@link ParameterDriver}.
  * <p>
@@ -33,7 +34,7 @@ public interface DSGenerator {
     /** Get the parameters selected for estimation.
      * @return parameters selected for estimation
      */
-    List<ParameterDriver> getSelected();
+    ParameterDriversList getSelected();
 
     /** Generate a constant {@link DerivativeStructure}.
      * @param value value of the constant