diff --git a/src/index.js b/src/index.js
index 7b18517..5f4610c 100644
--- a/src/index.js
+++ b/src/index.js
@@ -41,19 +41,16 @@ const plugin = withOptions(
     }
 
     return {
-      // TODO: Custom extraction logic so a magic comment can be used instead inside templates
       safelist: [
+        "bg-gray-100",
         {
-          pattern: /bg-+/,
+          pattern: /^bg-\w+-(50|500)$/,
         },
         {
-          pattern: /border-+/,
+          pattern: /^border-\w+-(600)$/,
         },
         {
-          pattern: /grid-cols-+/,
-        },
-        {
-          pattern: /text-+/,
+          pattern: /^text-\w+-(600|700)$/,
         },
       ],
       theme: {