diff --git a/trussme/member.py b/trussme/member.py
index 241fd73..9549d99 100644
--- a/trussme/member.py
+++ b/trussme/member.py
@@ -48,7 +48,7 @@ def __init__(self, joint_a: int, joint_b: int):
         self.set_material("A36", update_props=False)
         self.set_parameters(t=0.002, r=0.02, update_props=True)
 
-    def set_shape(self, new_shape: Literal["pipe", "bar", "square", "box"], update_props: bool = True):
+    def set_shape(self, new_shape: typing.Literal["pipe", "bar", "square", "box"], update_props: bool = True):
         # Read and save hte shape name
         if self.shape_name_is_ok(new_shape):
             self.shape = new_shape