diff --git a/src/Azure.EntityServices.Tables/Extensions/IAsyncEnumerableExtensions.cs b/src/Azure.EntityServices.Tables/Extensions/IAsyncEnumerableExtensions.cs
index 2dcb1ef..3af61b1 100644
--- a/src/Azure.EntityServices.Tables/Extensions/IAsyncEnumerableExtensions.cs
+++ b/src/Azure.EntityServices.Tables/Extensions/IAsyncEnumerableExtensions.cs
@@ -128,14 +128,11 @@ public static async IAsyncEnumerable<IEnumerable<T>> TakeAsync<T>(this IAsyncEnu
                 {
                     yield return asyncEntity;
                 }
-                else 
-                if (took >= take)
+                else
                 {
-                    yield return asyncEntity.Take(asyncEntity.Count()- (took - take) );
-                    
-                }
-                else 
-                yield break;
+                    yield return asyncEntity.Take(asyncEntity.Count() - (took - take));
+                    yield break;
+                } 
             }
         }
     }