From 16f49419e993de19fbc79e4648b104a13cffd06f Mon Sep 17 00:00:00 2001
From: WrathfulSpatula <stranoj@gmail.com>
Date: Wed, 4 Sep 2024 11:53:26 -0400
Subject: [PATCH] Fix wire label exception

---
 pennylane_qrack/_version.py      | 2 +-
 pennylane_qrack/qrack_device.cpp | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

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 {