diff --git a/mod.ts b/mod.ts
index c8aee46..69cd7be 100644
--- a/mod.ts
+++ b/mod.ts
@@ -70,7 +70,7 @@ export class FeatureFlags<
         return this._store;
     }
 
-    get(flag: FeatureFlag<Schema>) {
+    get<Flag extends FeatureFlag<Schema>>(flag: Flag): FeatureFlagOption<Schema, Flag> {
         return this._store[flag];
     }