diff --git a/ImageTools/UnitTests/CryptTests/TripleDesTest.cs b/ImageTools/UnitTests/CryptTests/TripleDesTest.cs
@@ -1,4 +1,5 @@
-using FunctionLib.Cryptography;
+using System.Security.Cryptography;
+using FunctionLib.Cryptography;
 using FunctionLib.Enums;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 
@@ -9,12 +10,12 @@ public class TripleDesTest : SymmetricAlgorithmBaseTestClass
     {
         public override string Encrypt(string value, string password)
         {
-            return SymmetricAlgorithmBase.Encrypt(this, EncryptionMethod.TripleDES, value, password);
+            return SymmetricAlgorithmBase.Encrypt(this, typeof(TripleDESCryptoServiceProvider), value, password);
         }
 
         public override string Decrypt(string value, string password)
         {
-            return SymmetricAlgorithmBase.Decrypt(this, EncryptionMethod.TripleDES, value, password);
+            return SymmetricAlgorithmBase.Decrypt(this, typeof(TripleDESCryptoServiceProvider), value, password);
         }
     }
 }
diff --git a/ImageTools/UnitTests/CryptTests/TwofishTest.cs b/ImageTools/UnitTests/CryptTests/TwofishTest.cs
@@ -1,4 +1,5 @@
 using FunctionLib.Cryptography;
+using FunctionLib.Cryptography.Twofish;
 using FunctionLib.Enums;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 
@@ -9,12 +10,12 @@ public class TwofishTest : SymmetricAlgorithmBaseTestClass
     {
         public override string Encrypt(string value, string password)
         {
-            return SymmetricAlgorithmBase.Encrypt(this, EncryptionMethod.Twofish, value, password);
+            return SymmetricAlgorithmBase.Encrypt(this, typeof(Twofish), value, password);
         }
 
         public override string Decrypt(string value, string password)
         {
-            return SymmetricAlgorithmBase.Decrypt(this, EncryptionMethod.Twofish, value, password);
+            return SymmetricAlgorithmBase.Decrypt(this, typeof(Twofish), value, password);
         }
     }
 }