diff --git a/Runtime/UIParticle.cs b/Runtime/UIParticle.cs
index b2c079b..5c0e4f7 100644
--- a/Runtime/UIParticle.cs
+++ b/Runtime/UIParticle.cs
@@ -9,6 +9,7 @@ using UnityEngine.UI;
 using Random = UnityEngine.Random;
 
 [assembly: InternalsVisibleTo("Coffee.UIParticle.Editor")]
+[assembly: InternalsVisibleTo("Coffee.UIParticle.Editor.Tests")]
 [assembly: InternalsVisibleTo("Coffee.UIParticle.PerformanceDemo")]
 [assembly: InternalsVisibleTo("Coffee.UIParticle.Demo")]
 
diff --git a/Runtime/Utilities/ParticleSystemExtensions.cs b/Runtime/Utilities/ParticleSystemExtensions.cs
index 4bd5c69..7fec948 100644
--- a/Runtime/Utilities/ParticleSystemExtensions.cs
+++ b/Runtime/Utilities/ParticleSystemExtensions.cs
@@ -13,11 +13,7 @@ namespace Coffee.UIParticleInternal
         {
             if (s_TmpParticles.Length < size)
             {
-                while (s_TmpParticles.Length < size)
-                {
-                    size = Mathf.NextPowerOfTwo(size);
-                }
-
+                size = Mathf.NextPowerOfTwo(size);
                 s_TmpParticles = new ParticleSystem.Particle[size];
             }