diff --git a/src/Neo/Neo.csproj b/src/Neo/Neo.csproj
index 75541f9091..694f4e1c02 100644
--- a/src/Neo/Neo.csproj
+++ b/src/Neo/Neo.csproj
@@ -26,6 +26,7 @@
 
   <ItemGroup>
     <InternalsVisibleTo Include="Neo.SmartContract.Testing" />
+    <InternalsVisibleTo Include="Neo.SmartContract.TestEngine" />
   </ItemGroup>
 
 </Project>
diff --git a/src/Neo/Persistence/MemoryStoreProvider.cs b/src/Neo/Persistence/MemoryStoreProvider.cs
index 23276bce96..e72a13c899 100644
--- a/src/Neo/Persistence/MemoryStoreProvider.cs
+++ b/src/Neo/Persistence/MemoryStoreProvider.cs
@@ -11,7 +11,7 @@
 
 namespace Neo.Persistence
 {
-    internal class MemoryStoreProvider : IStoreProvider
+    public class MemoryStoreProvider : IStoreProvider
     {
         public string Name => nameof(MemoryStore);
         public IStore GetStore(string path) => new MemoryStore();
diff --git a/src/Neo/SmartContract/Native/NativeContract.cs b/src/Neo/SmartContract/Native/NativeContract.cs
index 6d5d105b7d..8d5155ff92 100644
--- a/src/Neo/SmartContract/Native/NativeContract.cs
+++ b/src/Neo/SmartContract/Native/NativeContract.cs
@@ -199,7 +199,7 @@ private NativeContractsCache.CacheEntry GetAllowedMethods(ProtocolSettings setti
         /// <param name="settings">The <see cref="ProtocolSettings"/> where the HardForks are configured.</param>
         /// <param name="index">Block index</param>
         /// <returns>The <see cref="ContractState"/>.</returns>
-        internal ContractState GetContractState(ProtocolSettings settings, uint index)
+        public ContractState GetContractState(ProtocolSettings settings, uint index)
         {
             // Get allowed methods and nef script
             NativeContractsCache.CacheEntry allowedMethods = GetAllowedMethods(settings, index);