diff --git a/packages/widget_toolkit/CHANGELOG.md b/packages/widget_toolkit/CHANGELOG.md
index 8f8156a1..43738fe6 100644
--- a/packages/widget_toolkit/CHANGELOG.md
+++ b/packages/widget_toolkit/CHANGELOG.md
@@ -1,3 +1,6 @@
+## [0.1.1]
+- Added `SmallButtonType.icon` type to `SmallButton` widget
+
 ## [0.1.0]
 - Fixed loading state of 'SmallButton' widget
 ### Breaking changes:
diff --git a/packages/widget_toolkit/example/android/build.gradle b/packages/widget_toolkit/example/android/build.gradle
index 58a8c74b..713d7f6e 100644
--- a/packages/widget_toolkit/example/android/build.gradle
+++ b/packages/widget_toolkit/example/android/build.gradle
@@ -26,6 +26,6 @@ subprojects {
     project.evaluationDependsOn(':app')
 }
 
-task clean(type: Delete) {
+tasks.register("clean", Delete) {
     delete rootProject.buildDir
 }
diff --git a/packages/widget_toolkit/example/lib/main.dart b/packages/widget_toolkit/example/lib/main.dart
index e8cc0486..5f4eedc9 100644
--- a/packages/widget_toolkit/example/lib/main.dart
+++ b/packages/widget_toolkit/example/lib/main.dart
@@ -244,9 +244,25 @@ class CommonComponentsPage extends StatelessWidget {
                   Padding(
                     padding: const EdgeInsets.symmetric(
                         vertical: 10, horizontal: 15),
-                    child: SmallButton(
-                      onPressed: () {},
-                      icon: Icons.home_work_outlined,
+                    child: Row(
+                      mainAxisAlignment: MainAxisAlignment.center,
+                      children: [
+                        SmallButton(
+                          onPressed: () {},
+                          type: SmallButtonType.icon,
+                          icon: Icons.home_work_outlined,
+                        ),
+                        SmallButton(
+                          onPressed: () {},
+                          type: SmallButtonType.outline,
+                          icon: Icons.home_work_outlined,
+                        ),
+                        SmallButton(
+                          onPressed: () {},
+                          type: SmallButtonType.filled,
+                          icon: Icons.home_work_outlined,
+                        ),
+                      ],
                     ),
                   ),
                 ],
diff --git a/packages/widget_toolkit/example/pubspec.lock b/packages/widget_toolkit/example/pubspec.lock
index 43829ab5..4c20b799 100644
--- a/packages/widget_toolkit/example/pubspec.lock
+++ b/packages/widget_toolkit/example/pubspec.lock
@@ -337,18 +337,18 @@ packages:
     dependency: transitive
     description:
       name: leak_tracker
-      sha256: "7f0df31977cb2c0b88585095d168e689669a2cc9b97c309665e3386f3e9d341a"
+      sha256: "3f87a60e8c63aecc975dda1ceedbc8f24de75f09e4856ea27daf8958f2f0ce05"
       url: "https://pub.dev"
     source: hosted
-    version: "10.0.4"
+    version: "10.0.5"
   leak_tracker_flutter_testing:
     dependency: transitive
     description:
       name: leak_tracker_flutter_testing
-      sha256: "06e98f569d004c1315b991ded39924b21af84cf14cc94791b8aea337d25b57f8"
+      sha256: "932549fb305594d82d7183ecd9fa93463e9914e1b67cacc34bc40906594a1806"
       url: "https://pub.dev"
     source: hosted
-    version: "3.0.3"
+    version: "3.0.5"
   leak_tracker_testing:
     dependency: transitive
     description:
@@ -385,18 +385,18 @@ packages:
     dependency: transitive
     description:
       name: material_color_utilities
-      sha256: "0e0a020085b65b6083975e499759762399b4475f766c21668c4ecca34ea74e5a"
+      sha256: f7142bb1154231d7ea5f96bc7bde4bda2a0945d2806bb11670e30b850d56bdec
       url: "https://pub.dev"
     source: hosted
-    version: "0.8.0"
+    version: "0.11.1"
   meta:
     dependency: transitive
     description:
       name: meta
-      sha256: "7687075e408b093f36e6bbf6c91878cc0d4cd10f409506f7bc996f68220b9136"
+      sha256: bdb68674043280c3428e9ec998512fb681678676b3c54e773629ffe74419f8c7
       url: "https://pub.dev"
     source: hosted
-    version: "1.12.0"
+    version: "1.15.0"
   mime:
     dependency: transitive
     description:
@@ -582,10 +582,10 @@ packages:
     dependency: transitive
     description:
       name: test_api
-      sha256: "9955ae474176f7ac8ee4e989dadfb411a58c30415bcfb648fa04b2b8a03afa7f"
+      sha256: "5b8a98dafc4d5c4c9c72d8b31ab2b23fc13422348d2997120294d3bac86b4ddb"
       url: "https://pub.dev"
     source: hosted
-    version: "0.7.0"
+    version: "0.7.2"
   theme_tailor_annotation:
     dependency: transitive
     description:
@@ -710,10 +710,10 @@ packages:
     dependency: transitive
     description:
       name: vm_service
-      sha256: "3923c89304b715fb1eb6423f017651664a03bf5f4b29983627c4da791f74a4ec"
+      sha256: f652077d0bdf60abe4c1f6377448e8655008eef28f128bc023f7b5e8dfeb48fc
       url: "https://pub.dev"
     source: hosted
-    version: "14.2.1"
+    version: "14.2.4"
   watcher:
     dependency: transitive
     description:
diff --git a/packages/widget_toolkit/lib/src/lib_ui_components/buttons/small_button.dart b/packages/widget_toolkit/lib/src/lib_ui_components/buttons/small_button.dart
index cbc2fb58..520a2ead 100644
--- a/packages/widget_toolkit/lib/src/lib_ui_components/buttons/small_button.dart
+++ b/packages/widget_toolkit/lib/src/lib_ui_components/buttons/small_button.dart
@@ -8,7 +8,7 @@ import '../sized_loading_indicator.dart';
 import 'button_color_style.dart';
 import 'button_state.dart';
 
-enum SmallButtonType { filled, outline }
+enum SmallButtonType { filled, outline, icon }
 
 class SmallButton extends StatelessWidget {
   final String? tooltip;
@@ -122,6 +122,16 @@ class SmallButton extends StatelessWidget {
           onPressed: onPressed,
           child: icon,
         );
+      case SmallButtonType.icon:
+        return IconButton(
+          style: IconButton.styleFrom(
+            padding: const EdgeInsets.all(0),
+            shape: const CircleBorder(),
+            fixedSize: const Size(48, 48),
+          ),
+          icon: icon,
+          onPressed: onPressed,
+        );
     }
   }
 
diff --git a/packages/widget_toolkit/pubspec.yaml b/packages/widget_toolkit/pubspec.yaml
index a9c90270..d17f911f 100644
--- a/packages/widget_toolkit/pubspec.yaml
+++ b/packages/widget_toolkit/pubspec.yaml
@@ -1,6 +1,6 @@
 name: widget_toolkit
 description: The Widget Toolkit package consists of several lightweight and customisable UI components that boost productivity and reduce the development time.
-version: 0.1.0
+version: 0.1.1
 repository: https://github.com/Prime-Holding/widget_toolkit/tree/master/packages/widget_toolkit
 issue_tracker: https://github.com/orgs/Prime-Holding/projects/6
 homepage: https://www.primeholding.com
diff --git a/packages/widget_toolkit/test/helpers/golden_helper.dart b/packages/widget_toolkit/test/helpers/golden_helper.dart
index cc59e5ca..4cfbd03a 100644
--- a/packages/widget_toolkit/test/helpers/golden_helper.dart
+++ b/packages/widget_toolkit/test/helpers/golden_helper.dart
@@ -228,12 +228,12 @@ void runGoldenBuilderTests(
             ...GlobalMaterialLocalizations.delegates,
           ],
           theme: theme == Themes.light
-              ? ThemeData(
+              ? ThemeData.light().copyWith(
                   extensions: <ThemeExtension<dynamic>>[
                     WidgetToolkitTheme.light(),
                   ],
                 )
-              : ThemeData(
+              : ThemeData.dark().copyWith(
                   extensions: <ThemeExtension<dynamic>>[
                     WidgetToolkitTheme.dark(),
                   ],
diff --git a/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_empty.png b/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_empty.png
index 01a3a97f..bfca6a22 100644
Binary files a/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_empty.png and b/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_empty.png differ
diff --git a/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_loading.png b/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_loading.png
index 93669e7c..5fbe372a 100644
Binary files a/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_loading.png and b/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_loading.png differ
diff --git a/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_success.png b/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_success.png
index ae4d3bd4..87e3c7ac 100644
Binary files a/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_success.png and b/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_success.png differ
diff --git a/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_widget.png b/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_widget.png
index b50f0f0a..6b955b23 100644
Binary files a/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_widget.png and b/packages/widget_toolkit/test/lib_edit_address/view/goldens/dark_theme/edit_address_widget.png differ
diff --git a/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_empty.png b/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_empty.png
index d43b11c2..7d6f7c49 100644
Binary files a/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_empty.png and b/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_empty.png differ
diff --git a/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_loading.png b/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_loading.png
index 141b8a54..7289e41f 100644
Binary files a/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_loading.png and b/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_loading.png differ
diff --git a/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_success.png b/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_success.png
index 83a32e07..2a54e6ce 100644
Binary files a/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_success.png and b/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_success.png differ
diff --git a/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_widget.png b/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_widget.png
index 525aab12..f6e295c7 100644
Binary files a/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_widget.png and b/packages/widget_toolkit/test/lib_edit_address/view/goldens/light_theme/edit_address_widget.png differ
diff --git a/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_empty.png b/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_empty.png
index 7d3fff04..35f6ad6b 100644
Binary files a/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_empty.png and b/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_empty.png differ
diff --git a/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_error.png b/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_error.png
index a67bcf96..f1acd959 100644
Binary files a/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_error.png and b/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_error.png differ
diff --git a/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_loading.png b/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_loading.png
index 3e2263cf..2c463380 100644
Binary files a/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_loading.png and b/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_loading.png differ
diff --git a/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_success.png b/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_success.png
index b8717de3..9c52aea6 100644
Binary files a/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_success.png and b/packages/widget_toolkit/test/lib_item_picker/view/goldens/dark_theme/item_picker_success.png differ
diff --git a/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_empty.png b/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_empty.png
index 2686f32a..609a9374 100644
Binary files a/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_empty.png and b/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_empty.png differ
diff --git a/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_error.png b/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_error.png
index cf8b95bd..3bf15501 100644
Binary files a/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_error.png and b/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_error.png differ
diff --git a/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_loading.png b/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_loading.png
index 3a024367..2b451c9e 100644
Binary files a/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_loading.png and b/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_loading.png differ
diff --git a/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_success.png b/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_success.png
index d0f8036e..6a820e45 100644
Binary files a/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_success.png and b/packages/widget_toolkit/test/lib_item_picker/view/goldens/light_theme/item_picker_success.png differ
diff --git a/packages/widget_toolkit/test/lib_language_picker/view/goldens/dark_theme/language_picker_success.png b/packages/widget_toolkit/test/lib_language_picker/view/goldens/dark_theme/language_picker_success.png
index 21908783..4ac6dcb3 100644
Binary files a/packages/widget_toolkit/test/lib_language_picker/view/goldens/dark_theme/language_picker_success.png and b/packages/widget_toolkit/test/lib_language_picker/view/goldens/dark_theme/language_picker_success.png differ
diff --git a/packages/widget_toolkit/test/lib_language_picker/view/goldens/light_theme/language_picker_success.png b/packages/widget_toolkit/test/lib_language_picker/view/goldens/light_theme/language_picker_success.png
index d2ff7025..55ce4d2c 100644
Binary files a/packages/widget_toolkit/test/lib_language_picker/view/goldens/light_theme/language_picker_success.png and b/packages/widget_toolkit/test/lib_language_picker/view/goldens/light_theme/language_picker_success.png differ
diff --git a/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_empty.png b/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_empty.png
index 76e28464..67e9aaac 100644
Binary files a/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_empty.png and b/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_empty.png differ
diff --git a/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_error.png b/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_error.png
index 0d80f53f..75c86453 100644
Binary files a/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_error.png and b/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_error.png differ
diff --git a/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_loading.png b/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_loading.png
index 63dea242..79d210c1 100644
Binary files a/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_loading.png and b/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_loading.png differ
diff --git a/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_success.png b/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_success.png
index 6741746b..7c559be8 100644
Binary files a/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_success.png and b/packages/widget_toolkit/test/lib_search_picker/view/goldens/dark_theme/search_picker_success.png differ
diff --git a/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_empty.png b/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_empty.png
index a7096362..5301d80f 100644
Binary files a/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_empty.png and b/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_empty.png differ
diff --git a/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_error.png b/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_error.png
index afe16718..83f53a64 100644
Binary files a/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_error.png and b/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_error.png differ
diff --git a/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_loading.png b/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_loading.png
index f47373de..4617e35d 100644
Binary files a/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_loading.png and b/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_loading.png differ
diff --git a/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_success.png b/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_success.png
index 42f60cb9..cefbdaec 100644
Binary files a/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_success.png and b/packages/widget_toolkit/test/lib_search_picker/view/goldens/light_theme/search_picker_success.png differ
diff --git a/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/dark_theme/text_field_dialog_error.png b/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/dark_theme/text_field_dialog_error.png
index 7ec4b04e..210f04ac 100644
Binary files a/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/dark_theme/text_field_dialog_error.png and b/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/dark_theme/text_field_dialog_error.png differ
diff --git a/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/dark_theme/text_field_dialog_loading.png b/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/dark_theme/text_field_dialog_loading.png
index 755a7028..13f4ccc5 100644
Binary files a/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/dark_theme/text_field_dialog_loading.png and b/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/dark_theme/text_field_dialog_loading.png differ
diff --git a/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/light_theme/text_field_dialog_error.png b/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/light_theme/text_field_dialog_error.png
index 60c9575d..bfadbdc3 100644
Binary files a/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/light_theme/text_field_dialog_error.png and b/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/light_theme/text_field_dialog_error.png differ
diff --git a/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/light_theme/text_field_dialog_loading.png b/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/light_theme/text_field_dialog_loading.png
index 9f2c20bd..f78a1cd9 100644
Binary files a/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/light_theme/text_field_dialog_loading.png and b/packages/widget_toolkit/test/lib_text_field_dialog/view/goldens/light_theme/text_field_dialog_loading.png differ
diff --git a/packages/widget_toolkit/test/lib_ui_components/buttons/factory/small_button_factory.dart b/packages/widget_toolkit/test/lib_ui_components/buttons/factory/small_button_factory.dart
index 37ac9f45..ecb0e44f 100644
--- a/packages/widget_toolkit/test/lib_ui_components/buttons/factory/small_button_factory.dart
+++ b/packages/widget_toolkit/test/lib_ui_components/buttons/factory/small_button_factory.dart
@@ -4,10 +4,12 @@ import 'package:widget_toolkit/ui_components.dart';
 Widget smallButtonFactory({
   required ButtonStateModel state,
   required SmallButtonType type,
+  ButtonColorStyle? colorStyle,
 }) =>
     SmallButton(
       onPressed: () {},
       icon: Icons.add,
       state: state,
       type: type,
+      colorStyle: colorStyle,
     );
diff --git a/packages/widget_toolkit/test/lib_ui_components/buttons/view/goldens/dark_theme/small_button_golden_tests.png b/packages/widget_toolkit/test/lib_ui_components/buttons/view/goldens/dark_theme/small_button_golden_tests.png
index 74f05b8a..7e405792 100644
Binary files a/packages/widget_toolkit/test/lib_ui_components/buttons/view/goldens/dark_theme/small_button_golden_tests.png and b/packages/widget_toolkit/test/lib_ui_components/buttons/view/goldens/dark_theme/small_button_golden_tests.png differ
diff --git a/packages/widget_toolkit/test/lib_ui_components/buttons/view/goldens/light_theme/small_button_golden_tests.png b/packages/widget_toolkit/test/lib_ui_components/buttons/view/goldens/light_theme/small_button_golden_tests.png
index ba693306..d5d22a80 100644
Binary files a/packages/widget_toolkit/test/lib_ui_components/buttons/view/goldens/light_theme/small_button_golden_tests.png and b/packages/widget_toolkit/test/lib_ui_components/buttons/view/goldens/light_theme/small_button_golden_tests.png differ
diff --git a/packages/widget_toolkit/test/lib_ui_components/buttons/view/small_button_test.dart b/packages/widget_toolkit/test/lib_ui_components/buttons/view/small_button_test.dart
index f58498b0..89eaf558 100644
--- a/packages/widget_toolkit/test/lib_ui_components/buttons/view/small_button_test.dart
+++ b/packages/widget_toolkit/test/lib_ui_components/buttons/view/small_button_test.dart
@@ -8,7 +8,7 @@ import '../factory/small_button_factory.dart';
 void main() {
   runGoldenBuilderTests(
     'small_button_golden_tests',
-    surfaceSize: const Size(700, 350),
+    surfaceSize: const Size(700, 500),
     matcherCustomPump: (widget) =>
         widget.pump(const Duration(milliseconds: 350)),
     builder: GoldenBuilder.grid(
@@ -46,6 +46,22 @@ void main() {
       ..addScenario(
           'pressed outlined',
           smallButtonFactory(
-              type: SmallButtonType.outline, state: ButtonStateModel.pressed)),
+              type: SmallButtonType.outline, state: ButtonStateModel.pressed))
+      ..addScenario(
+          'loading icon',
+          smallButtonFactory(
+              type: SmallButtonType.icon, state: ButtonStateModel.loading))
+      ..addScenario(
+          'enabled icon',
+          smallButtonFactory(
+              type: SmallButtonType.icon, state: ButtonStateModel.enabled))
+      ..addScenario(
+          'disabled icon',
+          smallButtonFactory(
+              type: SmallButtonType.icon, state: ButtonStateModel.disabled))
+      ..addScenario(
+          'pressed icon',
+          smallButtonFactory(
+              type: SmallButtonType.icon, state: ButtonStateModel.pressed)),
   );
 }