diff --git a/examples/demo/demo.py b/examples/demo/demo.py
index 1edb469c..12eccb97 100644
--- a/examples/demo/demo.py
+++ b/examples/demo/demo.py
@@ -1,4 +1,5 @@
 import sys
+from unittest import mock
 
 from dependency_injector import containers, providers
 from dependency_injector.wiring import Provide
@@ -35,3 +36,6 @@ if __name__ == '__main__':
     container.wire(modules=[sys.modules[__name__]])
 
     main()
+
+    with container.api_client.override(mock.Mock()):
+        main()
diff --git a/examples/demo/test.py b/examples/demo/test.py
deleted file mode 100644
index ce5e0b6c..00000000
--- a/examples/demo/test.py
+++ /dev/null
@@ -1,11 +0,0 @@
-from unittest import mock
-
-import demo
-
-
-if __name__ == '__main__':
-    container = demo.Container()
-    container.wire(modules=[demo])
-
-    with container.api_client.override(mock.Mock()):
-        demo.main()