diff --git a/mobile/test/medium/repositories/timeline_repository_test.dart b/mobile/test/medium/repositories/timeline_repository_test.dart
index 8588a7c4f2..6fecf0e26d 100644
--- a/mobile/test/medium/repositories/timeline_repository_test.dart
+++ b/mobile/test/medium/repositories/timeline_repository_test.dart
@@ -1,6 +1,5 @@
 import 'package:flutter_test/flutter_test.dart';
 import 'package:immich_mobile/domain/models/asset/base_asset.model.dart';
-import 'package:immich_mobile/domain/models/timeline.model.dart';
 import 'package:immich_mobile/infrastructure/repositories/timeline.repository.dart';
 import 'package:intl/date_symbol_data_local.dart';
 
@@ -77,7 +76,7 @@ void main() {
       final user = await ctx.newUser();
       final asset = await ctx.newRemoteAsset(ownerId: user.id, livePhotoVideoId: 'motion-photo-1');
 
-      final assets = await sut.main([user.id], .day, AssetOriginFilter.all).assetSource(0, 10);
+      final assets = await sut.main([user.id], .day, .all).assetSource(0, 10);
 
       expect(assets, hasLength(1));
       final remote = assets.single as RemoteAsset;
@@ -93,7 +92,7 @@ void main() {
       final asset = await ctx.newRemoteAsset(ownerId: user.id, checksum: checksum, livePhotoVideoId: 'motion-photo-2');
       final local = await ctx.newLocalAsset(checksum: checksum);
 
-      final assets = await sut.main([user.id], .day, AssetOriginFilter.all).assetSource(0, 10);
+      final assets = await sut.main([user.id], .day, .all).assetSource(0, 10);
 
       expect(assets, hasLength(1));
       final remote = assets.single as RemoteAsset;