diff --git a/rllib/utils/spaces/simplex.py b/rllib/utils/spaces/simplex.py
index 13e0442ff..4f7d97676 100644
--- a/rllib/utils/spaces/simplex.py
+++ b/rllib/utils/spaces/simplex.py
@@ -23,7 +23,6 @@ class Simplex(gym.Space):
 
     def __init__(self, shape, concentration=None, dtype=np.float32):
         assert type(shape) in [tuple, list]
-        assert len(shape) == 2, ""
 
         super().__init__(shape, dtype)
         self.dim = self.shape[-1]