diff --git a/.github/CHANGELOG.md b/.github/CHANGELOG.md
index b12135199a..200fb0b7f8 100644
--- a/.github/CHANGELOG.md
+++ b/.github/CHANGELOG.md
@@ -16,6 +16,9 @@
 
 ### Bug fixes
 
+* Test updates to reflect new additions to PennyLane.
+[(#318)](https://github.com/PennyLaneAI/pennylane-lightning/pull/318)
+
 ### Contributors
 
 This release contains contributions from (in alphabetical order):
diff --git a/pennylane_lightning/_version.py b/pennylane_lightning/_version.py
index 9fc7d267e8..2061c816d4 100644
--- a/pennylane_lightning/_version.py
+++ b/pennylane_lightning/_version.py
@@ -16,4 +16,4 @@
    Version number (major.minor.patch[-label])
 """
 
-__version__ = "0.25.0-dev3"
+__version__ = "0.25.0-dev4"
diff --git a/tests/test_apply.py b/tests/test_apply.py
index e4f075348d..abe408ebc3 100644
--- a/tests/test_apply.py
+++ b/tests/test_apply.py
@@ -506,7 +506,10 @@ def test_apply_errors_qubit_state_vector(self, qubit_device_2_wires):
         with pytest.raises(ValueError, match="Sum of amplitudes-squared does not equal one."):
             qubit_device_2_wires.apply([qml.QubitStateVector(np.array([1, -1]), wires=[0])])
 
-        with pytest.raises(ValueError, match=r"State vector must be of length 2\*\*wires."):
+        with pytest.raises(
+            ValueError,
+            match=r"State vector must have shape \(2\*\*wires,\) or \(batch_size, 2\*\*wires\).",
+        ):
             p = np.array([1, 0, 1, 1, 0]) / np.sqrt(3)
             qubit_device_2_wires.apply([qml.QubitStateVector(p, wires=[0, 1])])