diff --git a/pennylane_qrack/_version.py b/pennylane_qrack/_version.py
index 63b1400..ddc7c2d 100644
--- a/pennylane_qrack/_version.py
+++ b/pennylane_qrack/_version.py
@@ -16,4 +16,4 @@
    Version number (major.minor.patch[-label])
 """
 
-__version__ = "0.8.8"
+__version__ = "0.9.0"
diff --git a/pennylane_qrack/qrack_device.cpp b/pennylane_qrack/qrack_device.cpp
index ca0938b..21b3ffb 100644
--- a/pennylane_qrack/qrack_device.cpp
+++ b/pennylane_qrack/qrack_device.cpp
@@ -457,6 +457,7 @@ struct QrackDevice final : public Catalyst::Runtime::QuantumDevice {
     auto AllocateQubit() -> QubitIdType override {
         const QubitIdType label = qsim->GetQubitCount();
         qsim->Allocate(1U);
+        qubit_map[label] = (bitLenInt)label;
         return label;
     }
     auto AllocateQubits(size_t num_qubits) -> std::vector<QubitIdType> override {