diff --git a/lib/widgets/notification.dart b/lib/widgets/notification.dart
index 970de6d..4544797 100644
--- a/lib/widgets/notification.dart
+++ b/lib/widgets/notification.dart
@@ -38,9 +38,9 @@ class NotificationItem extends StatelessWidget {
     );
   }
 
-  Future<void> markAsRead(element) async {
+  Future<void> markAsRead(model.Notification element) async {
     if (authClient.client != null) {
-      var id = element['id'];
+      var id = element.id;
       var uri = Uri.parse('https://id.solsynth.dev/api/notifications/$id/read');
       await authClient.client!.put(uri);
     }