diff --git a/bladex/shaft/cylinder_shaft.py b/bladex/shaft/cylinder_shaft.py
index f424366..c99732c 100644
--- a/bladex/shaft/cylinder_shaft.py
+++ b/bladex/shaft/cylinder_shaft.py
@@ -22,9 +22,13 @@ def __init__(self, radius=1.0, height=1.0, orientation=None, origin=None):
 
         if orientation is None:
             self.orientation = [1.0, 0.0, 0.0]  # default orientation along X
+        else:
+            self.orientation = orientation
 
         if origin is None:
             self.origin = [0.0, 0.0, 0.0]  # default origin at (0,0,0)
+        else:
+            self.origin = origin
 
     def generate_solid(self):
         """