diff --git a/src/backend/InvenTree/InvenTree/management/commands/schema.py b/src/backend/InvenTree/InvenTree/management/commands/schema.py
index ceff5c4f40..25c9cc38c2 100644
--- a/src/backend/InvenTree/InvenTree/management/commands/schema.py
+++ b/src/backend/InvenTree/InvenTree/management/commands/schema.py
@@ -108,3 +108,5 @@ class Command(spectacular.Command):
         settings.SPECTACULAR_SETTINGS['APPEND_COMPONENTS'] = components
 
         super().handle(*args, **kwargs)
+
+        return 'done'
diff --git a/src/backend/InvenTree/InvenTree/test_commands.py b/src/backend/InvenTree/InvenTree/test_commands.py
new file mode 100644
index 0000000000..955a70c062
--- /dev/null
+++ b/src/backend/InvenTree/InvenTree/test_commands.py
@@ -0,0 +1,13 @@
+"""Testts for custom InvenTree management commands."""
+
+from django.core.management import call_command
+from django.test import TestCase
+
+
+class CommandTestCase(TestCase):
+    """Test case for custom management commands."""
+
+    def test_schema(self):
+        """Test the schema generation command."""
+        output = call_command('schema', verbosity=0)
+        self.assertEqual(output, 'done')