diff --git a/nmigen/test/test_hdl_ir.py b/nmigen/test/test_hdl_ir.py
index 2ad5eb2..439b791 100644
--- a/nmigen/test/test_hdl_ir.py
+++ b/nmigen/test/test_hdl_ir.py
@@ -18,9 +18,11 @@ class FragmentGetTestCase(FHDLTestCase):
                 msg="Object 'None' cannot be elaborated"):
             Fragment.get(None, platform=None)
 
-        with self.assertRaises(AttributeError,
-                msg="Object 'None' cannot be elaborated"):
-            Fragment.get(BadElaboratable(), platform=None)
+        with self.assertWarns(UserWarning,
+                msg=".elaborate() returned None; missing return statement?"):
+            with self.assertRaises(AttributeError,
+                    msg="Object 'None' cannot be elaborated"):
+                Fragment.get(BadElaboratable(), platform=None)
 
 
 class FragmentGeneratedTestCase(FHDLTestCase):