From 7e2784c79974ffe936b8d0f1180823bd87c36742 Mon Sep 17 00:00:00 2001
From: uncenter <47499684+uncenter@users.noreply.github.com>
Date: Sun, 2 Jun 2024 21:28:06 -0400
Subject: [PATCH] fix: use correct count flag variable

---
 src/ports.rs      | 2 +-
 src/userstyles.rs | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/ports.rs b/src/ports.rs
index 58b8aee..5dd51c8 100644
--- a/src/ports.rs
+++ b/src/ports.rs
@@ -40,7 +40,7 @@ pub fn query(command: Option<Query>, r#for: Option<String>, count: bool, get: Ke
 				.map(|port| get_key(port, options.get))
 				.collect::<Vec<_>>();
 
-			display_json_or_count(&result, count)?;
+			display_json_or_count(&result, options.count)?;
 		}
 		Some(Query::Has {
 			name,
diff --git a/src/userstyles.rs b/src/userstyles.rs
index 5484986..d522792 100644
--- a/src/userstyles.rs
+++ b/src/userstyles.rs
@@ -44,7 +44,7 @@ pub fn query(
 				.map(|userstyle| get_userstyle_key(userstyle, options.get))
 				.collect::<Vec<_>>();
 
-			display_json_or_count(&result, count)?;
+			display_json_or_count(&result, options.count)?;
 		}
 		Some(UserstylesQuery::Has {
 			name,