diff --git a/waffle/management/commands/waffle_flag.py b/waffle/management/commands/waffle_flag.py
index 3fca539a..72ec7073 100644
--- a/waffle/management/commands/waffle_flag.py
+++ b/waffle/management/commands/waffle_flag.py
@@ -5,6 +5,7 @@
 from django.core.management.base import BaseCommand, CommandError, CommandParser
 from django.db.models import Q
 
+from build.lib.waffle.models import AbstractBaseFlag
 from waffle import get_waffle_flag_model
 
 UserModel = get_user_model()
@@ -166,7 +167,7 @@ def handle(self, *args: Any, **options: Any) -> None:
 
         flag.save()
 
-    def log_flag_to_stdout(self, flag):
+    def log_flag_to_stdout(self, flag: AbstractBaseFlag) -> None:
         self.stdout.write(f'NAME: {flag.name}')
         self.stdout.write(f'SUPERUSERS: {flag.superusers}')
         self.stdout.write(f'EVERYONE: {flag.everyone}')