diff --git a/mpitests/test_native_mcm.py b/mpitests/test_native_mcm.py
index 4ca607e34..78bde9872 100644
--- a/mpitests/test_native_mcm.py
+++ b/mpitests/test_native_mcm.py
@@ -23,7 +23,7 @@
 
 
 def test_unspported_mid_measurement():
-    """Test unsupported mid_measurement for lightning.gpu-mpi."""
+    """Test unsupported mid_measurement for Lightning-GPU-MPI."""
     comm = MPI.COMM_WORLD
     dev = qml.device(device_name, wires=2, mpi=True, shots=1000)
     params = np.pi / 4 * np.ones(2)
@@ -38,6 +38,6 @@ def func(x, y):
     comm.Barrier()
 
     with pytest.raises(
-        qml.DeviceError, match="LightningGPU-MPI does not support Mid-circuit measurements."
+        qml.DeviceError, match="Lightning-GPU-MPI does not support Mid-circuit measurements."
     ):
         func(*params)
diff --git a/pennylane_lightning/lightning_gpu/lightning_gpu.py b/pennylane_lightning/lightning_gpu/lightning_gpu.py
index 4d34048fd..1601f2124 100644
--- a/pennylane_lightning/lightning_gpu/lightning_gpu.py
+++ b/pennylane_lightning/lightning_gpu/lightning_gpu.py
@@ -351,7 +351,7 @@ def __init__(  # pylint: disable=too-many-arguments
         # Creating the state vector
         self._mpi_handler = MPIHandler(mpi, mpi_buf_size, len(self.wires), c_dtype)
 
-        #TODO: Remove this line after the MPI support for Mid-Measurement is added
+        # TODO: Remove this line after the MPI support for Mid-Measurement is added
         self._use_mpi = mpi
 
         self._statevector = self.LightningStateVector(
@@ -513,7 +513,9 @@ def simulate(
         """
         if circuit.shots and (any(isinstance(op, MidMeasureMP) for op in circuit.operations)):
             if self._use_mpi:
-                raise qml.DeviceError("LightningGPU-MPI does not support Mid-circuit measurements.")
+                raise qml.DeviceError(
+                    "Lightning-GPU-MPI does not support Mid-circuit measurements."
+                )
 
             results = []
             aux_circ = QuantumScript(