diff --git a/Makefile b/Makefile
index 2d07df0..7b9dd90 100644
--- a/Makefile
+++ b/Makefile
@@ -27,6 +27,8 @@ $(VENV_BIN_ACTIVATE): requirements.remote.txt requirements.txt
 
 test: $(VENV_BIN_ACTIVATE) .venv/${FW_VERSION}_reMarkable2-${FW_DATA}.signed
 	@echo "[info] Running test"
+	. $(VENV_BIN_ACTIVATE); \
+	python test.py
 	if [ -d .venv/mnt ] && mountpoint -q .venv/mnt; then \
 	    umount -ql .venv/mnt; \
 	fi
diff --git a/test.py b/test.py
new file mode 100644
index 0000000..4c4d286
--- /dev/null
+++ b/test.py
@@ -0,0 +1,37 @@
+import sys
+import codexctl
+
+FAILED = False
+
+print("Testing set_server_config: ", end="")
+result = codexctl.set_server_config(
+    """
+[General]
+#REMARKABLE_RELEASE_APPID={98DA7DF2-4E3E-4744-9DE6-EC931886ABAB}
+#SERVER=https://updates.cloud.remarkable.engineering/service/update2
+#GROUP=Prod
+#PLATFORM=reMarkable2
+REMARKABLE_RELEASE_VERSION=3.9.5.2026
+""",
+    "test",
+)
+if (
+    result
+    == """
+[General]
+SERVER=test
+#REMARKABLE_RELEASE_APPID={98DA7DF2-4E3E-4744-9DE6-EC931886ABAB}
+#SERVER=https://updates.cloud.remarkable.engineering/service/update2
+#GROUP=Prod
+#PLATFORM=reMarkable2
+REMARKABLE_RELEASE_VERSION=3.9.5.2026
+"""
+):
+    FAILED = True
+    print("fail")
+
+else:
+    print("pass")
+
+if FAILED:
+    sys.exit(1)