1 files changed, 5 insertions, 4 deletions
diff --git a/crypto/src/cmp/ProtectedPkiMessageBuilder.cs b/crypto/src/cmp/ProtectedPkiMessageBuilder.cs
index 22a004669..e660f844a 100644
--- a/crypto/src/cmp/ProtectedPkiMessageBuilder.cs
+++ b/crypto/src/cmp/ProtectedPkiMessageBuilder.cs
@@ -1,11 +1,13 @@
 using System;
 using System.Collections;
+using Org.BouncyCastle.Asn1.Crmf;
 using Org.BouncyCastle.Asn1;
 using Org.BouncyCastle.Asn1.Cmp;
 using Org.BouncyCastle.Asn1.X509;
 using Org.BouncyCastle.Crypto;
 using Org.BouncyCastle.Crypto.Operators;
 using Org.BouncyCastle.X509;
+using Org.BouncyCastle.Crmf;
 
 namespace Org.BouncyCastle.Cmp
 {
@@ -153,14 +155,13 @@ namespace Org.BouncyCastle.Cmp
             signer.Stream.Write(encoded, 0, encoded.Length);
             Object result = signer.GetResult();
 
-
             if (result is DefaultSignatureResult)
             {
-                return ((DefaultSignatureResult) result).Collect();
+                return ((DefaultSignatureResult)result).Collect();
             }
-            else if (result is DefaultMacAndDigestResult)
+            else if (result is IBlockResult)
             {
-                return ((DefaultMacAndDigestResult) result).MacResult;
+                return ((IBlockResult)result).Collect();
             }
             else if (result is byte[])
             {