diff --git a/test/counter_presenter_test.dart b/test/counter_presenter_test.dart
index 5bbfb11..13983f2 100644
--- a/test/counter_presenter_test.dart
+++ b/test/counter_presenter_test.dart
@@ -1,14 +1,16 @@
 import 'package:async/async.dart';
 import 'package:counter_with_onion_architecture/core/application_services/counter_presenter.dart';
 import 'package:counter_with_onion_architecture/core/domain/model/counter.dart';
+import 'package:counter_with_onion_architecture/core/domain/services/increment_counter.dart';
 import 'package:counter_with_onion_architecture/core/domain/services/increment_counter_fake_impl.dart';
+import 'package:counter_with_onion_architecture/infrastructure/counter_data_source.dart';
 import 'package:counter_with_onion_architecture/infrastructure/fake_counter_data_source.dart';
 import 'package:flutter_test/flutter_test.dart';
 
 void main() {
   test('CounterPresenter should update the counter', () async {
-    final FakeCounterDataSource dataSource = FakeCounterDataSource();
-    final IncrementCounterFakeImpl incrementCounter = IncrementCounterFakeImpl(
+    final CounterDataSource dataSource = FakeCounterDataSource();
+    final IncrementCounter incrementCounter = IncrementCounterFakeImpl(
       dataSource,
     );
     final CounterPresenter presenter = CounterPresenter(incrementCounter);
diff --git a/test/increment_counter_test.dart b/test/increment_counter_test.dart
index 5bbfb11..13983f2 100644
--- a/test/increment_counter_test.dart
+++ b/test/increment_counter_test.dart
@@ -1,14 +1,16 @@
 import 'package:async/async.dart';
 import 'package:counter_with_onion_architecture/core/application_services/counter_presenter.dart';
 import 'package:counter_with_onion_architecture/core/domain/model/counter.dart';
+import 'package:counter_with_onion_architecture/core/domain/services/increment_counter.dart';
 import 'package:counter_with_onion_architecture/core/domain/services/increment_counter_fake_impl.dart';
+import 'package:counter_with_onion_architecture/infrastructure/counter_data_source.dart';
 import 'package:counter_with_onion_architecture/infrastructure/fake_counter_data_source.dart';
 import 'package:flutter_test/flutter_test.dart';
 
 void main() {
   test('CounterPresenter should update the counter', () async {
-    final FakeCounterDataSource dataSource = FakeCounterDataSource();
-    final IncrementCounterFakeImpl incrementCounter = IncrementCounterFakeImpl(
+    final CounterDataSource dataSource = FakeCounterDataSource();
+    final IncrementCounter incrementCounter = IncrementCounterFakeImpl(
       dataSource,
     );
     final CounterPresenter presenter = CounterPresenter(incrementCounter);