diff --git a/lib/codegen/swift/urlsession.dart b/lib/codegen/swift/urlsession.dart
index c66964d9..25cce35a 100644
--- a/lib/codegen/swift/urlsession.dart
+++ b/lib/codegen/swift/urlsession.dart
@@ -1,6 +1,4 @@
-import 'package:apidash/consts.dart';
-import 'package:apidash/models/models.dart';
-import 'package:apidash/utils/utils.dart' show getValidRequestUri;
+import 'package:apidash_core/apidash_core.dart';
 import 'package:jinja/jinja.dart' as jj;
 import 'package:path/path.dart' as path;