diff --git a/lua/lualine/components/diff/init.lua b/lua/lualine/components/diff/init.lua
index 917549c..6ec33e5 100644
--- a/lua/lualine/components/diff/init.lua
+++ b/lua/lualine/components/diff/init.lua
@@ -14,13 +14,25 @@ local default_options = {
   symbols = { added = '+', modified = '~', removed = '-' },
   diff_color = {
     added = {
-      fg = modules.utils.extract_highlight_colors('DiffAdd', 'fg') or '#90ee90',
+      fg = modules.utils.extract_color_from_hllist(
+        'fg',
+        { 'GitSignsAdd', 'GitGutterAdd', 'DiffAdded', 'DiffAdd' },
+        '#90ee90'
+      ),
     },
     modified = {
-      fg = modules.utils.extract_highlight_colors('DiffChange', 'fg') or '#f0e130',
+      fg = modules.utils.extract_color_from_hllist(
+        'fg',
+        { 'GitSignsChange', 'GitGutterChange', 'DiffChanged', 'DiffChange' },
+        '#f0e130'
+      ),
     },
     removed = {
-      fg = modules.utils.extract_highlight_colors('DiffDelete', 'fg') or '#ff0038',
+      fg = modules.utils.extract_color_from_hllist(
+        'fg',
+        { 'GitSignsDelete', 'GitGutterDelete', 'DiffRemoved', 'DiffDelete' },
+        '#ff0038'
+      ),
     },
   },
 }