diff --git a/gamelauncher_plugins.py b/gamelauncher_plugins.py
index 4c176e8..f8c33bb 100644
--- a/gamelauncher_plugins.py
+++ b/gamelauncher_plugins.py
@@ -3,7 +3,7 @@ from pathlib import Path
 from typing import Dict, Set
 
 
-def enable_steam_game_drive(env: Dict[str, str]):
+def enable_steam_game_drive(env: Dict[str, str]) -> Dict[str, str]:
     """Enable Steam Game Drive functionality.
 
     Expects STEAM_COMPAT_INSTALL_PATH to be set