Update elastica/typing.py
@@ -4,4 +4,4 @@ from elastica.rigidbody import RigidBodyBase
from typing import Type, Union
RodType = Type[RodBase]
-SystemType = Union[Type[RodBase], Type[RigidBodyBase]]
+SystemType = Union[RodType, Type[RigidBodyBase]]
