1 files changed, 5 insertions, 3 deletions
diff --git a/src/qregisters.h b/src/qregisters.h
index 1c55066..48b023d 100644
--- a/src/qregisters.h
+++ b/src/qregisters.h
@@ -466,12 +466,14 @@ private:
 	State *got_file(const gchar *filename);
 };
 
-class StateQueryQReg : public StateExpectQReg {
+class StateQueryQReg : public State {
+	QRegSpecMachine machine;
+
 public:
-	StateQueryQReg() : StateExpectQReg(QREG_OPTIONAL) {}
+	StateQueryQReg();
 
 private:
-	State *got_register(QRegister *reg);
+	State *custom(gchar chr);
 };
 
 class StateCtlUCommand : public StateExpectQReg {