diff --git a/test/game/view/widgets/bonus_animation_test.dart b/test/game/view/widgets/bonus_animation_test.dart
index 2284ca8d..52c1b3d8 100644
--- a/test/game/view/widgets/bonus_animation_test.dart
+++ b/test/game/view/widgets/bonus_animation_test.dart
@@ -1,9 +1,5 @@
 // ignore_for_file: invalid_use_of_protected_member
 
-import 'dart:typed_data';
-
-import 'package:flame/assets.dart';
-import 'package:flame/flame.dart';
 import 'package:flutter/material.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:mocktail/mocktail.dart';
@@ -13,8 +9,6 @@ import 'package:pinball_flame/pinball_flame.dart';
 
 import '../../../helpers/helpers.dart';
 
-class _MockImages extends Mock implements Images {}
-
 class _MockCallback extends Mock {
   void call();
 }
@@ -24,13 +18,7 @@ void main() {
   const animationDuration = 6;
 
   setUp(() async {
-    // TODO(arturplaczek): need to find for a better solution for loading image
-    // or use original images from BonusAnimation.loadAssets()
-    final image = await decodeImageFromList(Uint8List.fromList(fakeImage));
-    final images = _MockImages();
-    when(() => images.fromCache(any())).thenReturn(image);
-    when(() => images.load(any())).thenAnswer((_) => Future.value(image));
-    Flame.images = images;
+    await mockFlameImages();
   });
 
   group('loads SpriteAnimationWidget correctly for', () {
diff --git a/test/game/view/widgets/game_hud_test.dart b/test/game/view/widgets/game_hud_test.dart
index f8be70c2..75fa7439 100644
--- a/test/game/view/widgets/game_hud_test.dart
+++ b/test/game/view/widgets/game_hud_test.dart
@@ -1,11 +1,8 @@
 // ignore_for_file: prefer_const_constructors
 
 import 'dart:async';
-import 'dart:typed_data';
 
 import 'package:bloc_test/bloc_test.dart';
-import 'package:flame/assets.dart';
-import 'package:flame/flame.dart';
 import 'package:flutter/material.dart';
 import 'package:flutter_bloc/flutter_bloc.dart';
 import 'package:flutter_localizations/flutter_localizations.dart';
@@ -18,8 +15,6 @@ import 'package:pinball_ui/pinball_ui.dart';
 
 import '../../../helpers/helpers.dart';
 
-class _MockImages extends Mock implements Images {}
-
 class _MockGameBloc extends Mock implements GameBloc {}
 
 void main() {
@@ -34,15 +29,9 @@ void main() {
     );
 
     setUp(() async {
-      gameBloc = _MockGameBloc();
+      await mockFlameImages();
 
-      // TODO(arturplaczek): need to find for a better solution for loading
-      // image or use original images from BonusAnimation.loadAssets()
-      final image = await decodeImageFromList(Uint8List.fromList(fakeImage));
-      final images = _MockImages();
-      when(() => images.fromCache(any())).thenReturn(image);
-      when(() => images.load(any())).thenAnswer((_) => Future.value(image));
-      Flame.images = images;
+      gameBloc = _MockGameBloc();
 
       whenListen(
         gameBloc,