diff --git a/bofire/strategies/random.py b/bofire/strategies/random.py
index 6297d06fb..98e806a37 100644
--- a/bofire/strategies/random.py
+++ b/bofire/strategies/random.py
@@ -331,4 +331,4 @@ def _sample_from_polytope(
         for key, value in fixed_features.items():
             samples[key] = value
 
-        return samples
+        return samples[domain.inputs.get_keys()]