diff --git a/src/ui.rs b/src/ui.rs
index 2f5e01a..b4f246a 100644
--- a/src/ui.rs
+++ b/src/ui.rs
@@ -94,11 +94,12 @@ fn add_ui_text(added: Query<Entity, Added<Text>>, mut commands: Commands) {
     fn pressed(
         trigger: Trigger<Pointer<Pressed>>,
         mut query: Query<(&mut TextColor, &ChildOf)>,
+        buttons: Query<Entity, With<Button>>,
         style: Res<Style>,
     ) {
         // ONLY DO THIS IF CHILD OF BUTTON
         if let Ok((mut tc, ChildOf(p))) = query.get_mut(trigger.target()) {
-            if buttons.contains(p) {
+            if buttons.contains(*p) {
                 debug!("pointer pressed {:?}", trigger.target());
                 tc.0 = style.primary;
             }
@@ -113,7 +114,7 @@ fn add_ui_text(added: Query<Entity, Added<Text>>, mut commands: Commands) {
     ) {
         // ONLY DO THIS IF CHILD OF BUTTON
         if let Ok((mut tc, ChildOf(p))) = query.get_mut(trigger.target()) {
-            if buttons.contains(p) {
+            if buttons.contains(*p) {
                 debug!("pointer released {:?}", trigger.target());
                 tc.0 = style.secondary;
             }