From ea98495e9cff317b5313b22e23f32fc0ef385a9c Mon Sep 17 00:00:00 2001
From: Davide Oberto <davide.oberto@polito.it>
Date: Thu, 18 Dec 2025 09:11:10 +0100
Subject: [PATCH] Fixed when optional arguments in init are specified

---
 bladex/shaft/cylinder_shaft.py | 4 ++++
 1 file changed, 4 insertions(+)

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):
         """