1 files changed, 10 insertions, 3 deletions
diff --git a/macros/ui/local-shared-attribute.rs b/macros/ui/local-shared-attribute.rs
index 1ccce4ad..c594b5f6 100644
--- a/macros/ui/local-shared-attribute.rs
+++ b/macros/ui/local-shared-attribute.rs
@@ -2,13 +2,20 @@
 
 #[rtic_macros::mock_app(device = mock)]
 mod app {
+    #[shared]
+    struct Shared {}
+
+    #[local]
+    struct Local {}
+
+    #[init]
+    fn init(_: init::Context) -> (Shared, Local) {}
+
     #[task(local = [
         #[test]
         a: u32 = 0, // Ok
         #[test]
         b, // Error
     ])]
-    fn foo(_: foo::Context) {
-
-    }
+    fn foo(_: foo::Context) {}
 }