diff --git a/tests/test_context.py b/tests/test_context.py
index 8a0412e..db03c32 100644
--- a/tests/test_context.py
+++ b/tests/test_context.py
@@ -63,6 +63,13 @@ def test_ctx_load_module(self):
             mod = ctx.load_module("yolo-system")
             self.assertIsInstance(mod, Module)
 
+    def test_ctx_load_module_with_features(self):
+        with Context(YANG_DIR) as ctx:
+            mod = ctx.load_module("yolo-system", None, ["*"])
+            self.assertIsInstance(mod, Module)
+            for f in list(mod.features()):
+                self.assertTrue(f.state())
+
     def test_ctx_get_module(self):
         with Context(YANG_DIR) as ctx:
             ctx.load_module("yolo-system")