1 files changed, 4 insertions, 4 deletions
diff --git a/lib/aes.go b/lib/aes.go
index 396ee5c..8c4dd26 100644
--- a/lib/aes.go
+++ b/lib/aes.go
@@ -23,7 +23,7 @@ func AESEncryptCBC(plain, key, iv []byte) []byte {
 	return output
 }
 
-func AESDecryptCBC(cipher, key, iv []byte) []byte {
+func AESDecryptCBC(cipher, key, iv []byte) ([]byte, error) {
 	iter := len(cipher) / 16
 
 	lc := iv
@@ -38,9 +38,9 @@ func AESDecryptCBC(cipher, key, iv []byte) []byte {
 	}
 
 	// Undo padding
-	output = Pkcs7PaddingUndo(output)
+	output, err := Pkcs7PaddingUndo(output)
 
-	return output
+	return output, err
 }
 
 func AESEncryptECB(plain, key []byte) []byte {
@@ -72,7 +72,7 @@ func AESDecryptECB(cipher, key []byte) []byte {
 	}
 
 	// Undo padding
-	output = Pkcs7PaddingUndo(output)
+	output, _ = Pkcs7PaddingUndo(output)
 
 	return output
 }