diff --git a/src/Agent/Responses/QueryResponse.cs b/src/Agent/Responses/QueryResponse.cs
index 7ef50385..4cfcc9d8 100644
--- a/src/Agent/Responses/QueryResponse.cs
+++ b/src/Agent/Responses/QueryResponse.cs
@@ -120,10 +120,10 @@ internal static QueryResponse ReadCbor(CborReader reader)
 						reader.ReadEndMap();
 						break;
 					case "reject_code":
-						CborTag codeType = reader.PeekTag();
-						switch (codeType)
+						CborReaderState state = reader.PeekState();
+						switch (state)
 						{
-							case CborTag.UnsignedBigNum:
+							case CborReaderState.UnsignedInteger:
 								rejectCode = reader.ReadUInt64();
 								break;
 							default: