diff --git a/frontools/cli.py b/frontools/cli.py
index b5a1583..29a2bb4 100644
--- a/frontools/cli.py
+++ b/frontools/cli.py
@@ -6,7 +6,7 @@ from typing import Any, Optional
 
 from click import Context as ClickContext
 from click import Path as PathArgument
-from click import argument, group, option, pass_context
+from click import argument, group, option, pass_context, pass_obj
 
 from frontools.config import Config
 from frontools.css import css_diff
@@ -50,11 +50,10 @@ async def main(
 
 @main.command(name="css-diff")
 @argument("right_source", type=str)
-@pass_context
+@pass_obj
 @_async_command
-async def css_diff_cli(ctx: ClickContext, right_source: str) -> None:
+async def css_diff_cli(config: Config, right_source: str) -> None:
     """Diff CSS"""
-    config: Config = ctx.obj
     for _, site in config.sites:
         for site_url in site.urls:
             await css_diff(
@@ -67,15 +66,14 @@ async def css_diff_cli(ctx: ClickContext, right_source: str) -> None:
 @option("-o", "--output-directory", type=PathArgument(), default=None)
 @option("-r", "--resolution", type=str, default=None)
 @_async_command
-@pass_context
+@pass_obj
 async def screenshot_diff_cli(
-    ctx: ClickContext,
+    config: Config,
     source: str,
     output_directory: Optional[str],
     resolution: Optional[str],
 ) -> None:
     """Generate screenshot diffs"""
-    config: Config = ctx.obj
     await screenshot_diff(config, source, output_directory, resolution=resolution)