diff --git a/lib/src/card_swiper.dart b/lib/src/card_swiper.dart
index b46f8fa..f307aba 100644
--- a/lib/src/card_swiper.dart
+++ b/lib/src/card_swiper.dart
@@ -1,6 +1,6 @@
 import 'dart:math';
 
-import 'package:flutter/material.dart';
+import 'package:flutter/widgets.dart';
 import 'package:flutter_card_swiper/src/card_swiper_controller.dart';
 import 'package:flutter_card_swiper/src/enums.dart';
 import 'package:flutter_card_swiper/src/typedefs.dart';
diff --git a/lib/src/card_swiper_controller.dart b/lib/src/card_swiper_controller.dart
index b6828a5..8e3e84d 100644
--- a/lib/src/card_swiper_controller.dart
+++ b/lib/src/card_swiper_controller.dart
@@ -1,4 +1,4 @@
-import 'package:flutter/widgets.dart';
+import 'package:flutter/foundation.dart';
 import 'package:flutter_card_swiper/src/enums.dart';
 
 //to call the swipe function from outside of the CardSwiper