diff --git a/pythonbindings/pyfluids-stubs/bindings/basics/__init__.pyi b/pythonbindings/pyfluids-stubs/bindings/basics/__init__.pyi
index a41b7934ca706dc0db5bd6188fee3150456e0cd9..a8b7ae9e7616d2cffe4eec1d525458286972ba48 100644
--- a/pythonbindings/pyfluids-stubs/bindings/basics/__init__.pyi
+++ b/pythonbindings/pyfluids-stubs/bindings/basics/__init__.pyi
@@ -36,6 +36,7 @@ from typing import ClassVar
 
 from typing import overload
 
+from pyfluids.bindings.basics import logger as logger
 class ConfigurationFile:
     def __init__(self) -> None: ...
     def contains(self, key: str) -> bool: ...
diff --git a/pythonbindings/pyfluids-stubs/bindings/gpu/__init__.pyi b/pythonbindings/pyfluids-stubs/bindings/gpu/__init__.pyi
index 36c2fea76713e980bb95eb6726d778de8c9a6583..41b019960026a2f9a7b0697dfd80770af12fa0d9 100644
--- a/pythonbindings/pyfluids-stubs/bindings/gpu/__init__.pyi
+++ b/pythonbindings/pyfluids-stubs/bindings/gpu/__init__.pyi
@@ -37,8 +37,9 @@ from typing import Any, Callable, ClassVar, List, Optional
 from typing import overload
 import numpy
 import pyfluids.bindings.basics
-import pyfluids.bindings.gpu.grid_generator as grid_generator
 
+from  pyfluids.bindings.gpu import grid_generator as grid_generator
+from pyfluids.bindings.gpu import probes as probes
 class ActuatorFarm(PreCollisionInteractor):
     def __init__(self, number_of_blades_per_turbine: int, density: float, number_of_nodes_per_blade: int, epsilon: float, level: int, delta_t: float, delta_x: float, use_host_arrays: bool) -> None: ...
     def add_turbine(self, posX: float, posY: float, posZ: float, diameter: float, omega: float, azimuth: float, yaw: float, bladeRadii: List[float]) -> None: ...