diff --git a/pythonbindings/pyfluids-stubs/bindings/gpu/__init__.pyi b/pythonbindings/pyfluids-stubs/bindings/gpu/__init__.pyi
index 36c2fea76713e980bb95eb6726d778de8c9a6583..64a598ee1974b089393b328566def02fb3600005 100644
--- a/pythonbindings/pyfluids-stubs/bindings/gpu/__init__.pyi
+++ b/pythonbindings/pyfluids-stubs/bindings/gpu/__init__.pyi
@@ -132,10 +132,10 @@ class BoundaryConditionFactory:
     def set_stress_boundary_condition(self, boundary_condition_type) -> None: ...
     def set_velocity_boundary_condition(self, boundary_condition_type) -> None: ...
 
-class Communicator:
+class MpiCommunicator:
     def __init__(self, *args, **kwargs) -> None: ...
     @staticmethod
-    def get_instance() -> Communicator: ...
+    def get_instance() -> MpiCommunicator: ...
     def get_number_of_process(self) -> int: ...
     def get_pid(self) -> int: ...
 
diff --git a/pythonbindings/src/gpu/submodules/communicator.cpp b/pythonbindings/src/gpu/submodules/communicator.cpp
index 3b0550e4347b52df2cbcda6706435718e8bce354..f17fa749461b522ffb3254cad7f9b6675d3de4fc 100644
--- a/pythonbindings/src/gpu/submodules/communicator.cpp
+++ b/pythonbindings/src/gpu/submodules/communicator.cpp
@@ -39,7 +39,7 @@ namespace communicator
 
     void makeModule(py::module_ &parentModule)
     {
-        py::class_<vf::gpu::MpiCommunicator, std::unique_ptr<vf::gpu::MpiCommunicator, py::nodelete>>(parentModule, "Communicator")
+        py::class_<vf::gpu::MpiCommunicator, std::unique_ptr<vf::gpu::MpiCommunicator, py::nodelete>>(parentModule, "MpiCommunicator")
         .def_static("get_instance", &vf::gpu::MpiCommunicator::getInstance, py::return_value_policy::reference)
         .def("get_number_of_process", &vf::gpu::MpiCommunicator::getNumberOfProcess)
         .def("get_pid", &vf::gpu::MpiCommunicator::getPID);