diff --git a/pyinfra/server/processor/processor.py b/pyinfra/server/processor/processor.py
index 4321011..80cd220 100644
--- a/pyinfra/server/processor/processor.py
+++ b/pyinfra/server/processor/processor.py
@@ -1,5 +1,5 @@
 from itertools import chain
-from typing import Union, Any, Iterable, Callable
+from typing import Union, Any
 
 from pyinfra.server.dispatcher.dispatcher import Nothing
 
diff --git a/test/unit_tests/server/__init__.py b/test/unit_tests/server/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/test/unit_tests/buffer_test.py b/test/unit_tests/server/buffer_test.py
similarity index 74%
rename from test/unit_tests/buffer_test.py
rename to test/unit_tests/server/buffer_test.py
index ad929fc..adba909 100644
--- a/test/unit_tests/buffer_test.py
+++ b/test/unit_tests/server/buffer_test.py
@@ -17,3 +17,10 @@ def test_buffer():
     reverse_buffer = bufferize(compose(reverse_buffer, list), buffer_size=3)
     ys = lmapcat(reverse_buffer, range(10))
     assert ys == [2, 1, 0, 5, 4, 3, 8, 7, 6]
+
+    def buffer_sum(xs):
+        return [sum(xs)]
+
+    buffer_sum = bufferize(buffer_sum, buffer_size=3)
+    ys = lmapcat(buffer_sum, range(10))
+    assert ys == [0, 0, 3, 0, 0, 12, 0, 0, 21, 0]
diff --git a/test/unit_tests/pipeline_test.py b/test/unit_tests/server/pipeline_test.py
similarity index 100%
rename from test/unit_tests/pipeline_test.py
rename to test/unit_tests/server/pipeline_test.py