diff --git a/lib/src/day_view/_internal_day_view_page.dart b/lib/src/day_view/_internal_day_view_page.dart
index 804ea239..d46be383 100644
--- a/lib/src/day_view/_internal_day_view_page.dart
+++ b/lib/src/day_view/_internal_day_view_page.dart
@@ -176,7 +176,7 @@ class InternalDayViewPage<T extends Object?> extends StatefulWidget {
   }) : super(key: key);
 
   @override
-  State<InternalDayViewPage> createState() => _InternalDayViewPageState();
+  _InternalDayViewPageState<T> createState() => _InternalDayViewPageState();
 }
 
 class _InternalDayViewPageState<T extends Object?>
diff --git a/lib/src/week_view/_internal_week_view_page.dart b/lib/src/week_view/_internal_week_view_page.dart
index ea2f570a..2603e890 100644
--- a/lib/src/week_view/_internal_week_view_page.dart
+++ b/lib/src/week_view/_internal_week_view_page.dart
@@ -211,7 +211,7 @@ class InternalWeekViewPage<T extends Object?> extends StatefulWidget {
   }) : super(key: key);
 
   @override
-  State<InternalWeekViewPage> createState() => _InternalWeekViewPageState();
+  _InternalWeekViewPageState<T> createState() => _InternalWeekViewPageState();
 }
 
 class _InternalWeekViewPageState<T extends Object?>