diff --git a/src/lib/flags.c b/src/lib/flags.c
index bcaf841..a78f492 100644
--- a/src/lib/flags.c
+++ b/src/lib/flags.c
@@ -74,6 +74,11 @@ void setFlags(inputFlags newFlags)
     flags = newFlags;
 }
 
+void setEncryptionOperation(encryptionOperation_type_cb newEncryptionOperation)
+{
+    encryptionOperation_cb = newEncryptionOperation;
+}
+
 //SOME OTHER SHIT
 int encryptionOperation(int text, int encryptedText)
 {