diff --git a/atmosphere/tests/e2e/test_operator.py b/atmosphere/tests/e2e/test_operator.py
index adc59e4..e4c9ccb 100644
--- a/atmosphere/tests/e2e/test_operator.py
+++ b/atmosphere/tests/e2e/test_operator.py
@@ -4,7 +4,7 @@
 
 import pykube
 import pytest
-import toml
+import tomli
 from jinja2 import Environment, FileSystemLoader
 from python_on_whales import docker
 from tenacity import Retrying, retry_if_exception_type, stop_after_delay, wait_fixed
@@ -28,7 +28,7 @@
         loader=FileSystemLoader("roles/atmosphere/templates"),
         extensions=["jinja2_base64_filters.Base64Filters"],
     )
-    env.filters["vexxhost.atmosphere.to_toml"] = toml.dumps
+    env.filters["vexxhost.atmosphere.to_toml"] = tomli.dumps
 
     args = {
         "atmosphere_image": docker_image,