Reorder classes

1 files changed, 2 insertions, 3 deletions

diff --git a/test/widget_test.dart b/test/widget_test.dart
index 11b1c70..58c70d5 100644
--- a/test/widget_test.dart
+++ b/test/widget_test.dart
@@ -5,15 +5,14 @@
 // gestures. You can also use WidgetTester to find child widgets in the widget
 // tree, read text, and verify that the values of widget properties are correct.
 
+import 'package:badge/badge_app.dart';
 import 'package:flutter/material.dart';
 import 'package:flutter_test/flutter_test.dart';
 
-import 'package:badge/main.dart';
-
 void main() {
   testWidgets('Counter increments smoke test', (WidgetTester tester) async {
     // Build our app and trigger a frame.
-    await tester.pumpWidget(const MyApp());
+    await tester.pumpWidget(const BadgeApp());
 
     // Verify that our counter starts at 0.
     expect(find.text('0'), findsOneWidget);