diff --git a/test/extensions/context_extensions_test.dart b/packages/apidash_design_system/test/extensions/context_extensions_test.dart
similarity index 98%
rename from test/extensions/context_extensions_test.dart
rename to packages/apidash_design_system/test/extensions/context_extensions_test.dart
index 282ec929..c49cf713 100644
--- a/test/extensions/context_extensions_test.dart
+++ b/packages/apidash_design_system/test/extensions/context_extensions_test.dart
@@ -1,7 +1,6 @@
 import 'package:apidash_design_system/apidash_design_system.dart';
 import 'package:flutter/material.dart';
 import 'package:flutter_test/flutter_test.dart';
-import 'package:apidash/extensions/context_extensions.dart';
 
 void main() {
   group('Testing MediaQueryExtension', () {
diff --git a/packages/apidash_design_system/test/checkbox_test.dart b/packages/apidash_design_system/test/widgets/checkbox_test.dart
similarity index 100%
rename from packages/apidash_design_system/test/checkbox_test.dart
rename to packages/apidash_design_system/test/widgets/checkbox_test.dart