From cef89eeee080470887058cff05ec1db52d4eef32 Mon Sep 17 00:00:00 2001
From: Taylor Holliday <taylor@audulus.com>
Date: Mon, 12 Jun 2023 11:24:22 -0700
Subject: [PATCH] Cleanup

---
 src/context.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/context.rs b/src/context.rs
index 6f2236f..90bf9b1 100644
--- a/src/context.rs
+++ b/src/context.rs
@@ -402,7 +402,7 @@ where
 {
     type Output = S;
 
-    fn index(&self, index: StateHandle<S>) -> &Self::Output {
+    fn index(&self, index: StateHandle<S>) -> &S {
         self.get(index)
     }
 }
@@ -411,7 +411,7 @@ impl<S> ops::IndexMut<StateHandle<S>> for Context
 where
     S: 'static,
 {
-    fn index_mut(&mut self, index: StateHandle<S>) -> &mut Self::Output {
+    fn index_mut(&mut self, index: StateHandle<S>) -> &mut S {
         self.get_mut(index)
     }
 }