diff --git a/codegen/src/function.rs b/codegen/src/function.rs
index 011ff650..89358274 100644
--- a/codegen/src/function.rs
+++ b/codegen/src/function.rs
@@ -1067,12 +1067,3 @@ mod generate_tests {
         assert_streams_eq(item_fn.generate(), expected_tokens);
     }
 }
-
-#[cfg(test)]
-mod ui_tests {
-    #[test]
-    fn all() {
-        let t = trybuild::TestCases::new();
-        t.compile_fail("ui_tests/*.rs");
-    }
-}
diff --git a/codegen/tests/ui_tests.rs b/codegen/tests/ui_tests.rs
new file mode 100644
index 00000000..ef7bf58f
--- /dev/null
+++ b/codegen/tests/ui_tests.rs
@@ -0,0 +1,8 @@
+#![cfg(test)]
+mod ui_tests {
+    #[test]
+    fn all() {
+        let t = trybuild::TestCases::new();
+        t.compile_fail("ui_tests/*.rs");
+    }
+}