diff --git a/.github/workflows/test_action.yml b/.github/workflows/test_action.yml
index 12088d1..6cd439b 100644
--- a/.github/workflows/test_action.yml
+++ b/.github/workflows/test_action.yml
@@ -14,9 +14,9 @@ jobs:
     steps:
     - uses: actions/checkout@v2
 
-    - uses: arthurpalves/coherent-swift@0.5.4.7
+    - uses: arthurpalves/coherent-swift@0.5.5
       with:
-        source: Sources
+        source: Sources/
         minimum_threshold: 60
         ignore_output_result: true
         only_changes: true
diff --git a/action.yml b/action.yml
index d3ba32d..5fab4d7 100644
--- a/action.yml
+++ b/action.yml
@@ -1,6 +1,9 @@
 name: 'coherent-swift'
 description: 'Measures cohesion in your Swift codebase'
 author: 'Arthur Alves <arthuralves.p@gmail.com>'
+branding:
+  icon: "code"
+  color: "green"
 
 inputs:
   source: