diff --git a/packages/palette_generator/test/palette_generator_test.dart b/packages/palette_generator/test/palette_generator_test.dart
index 7738a8e171..a7a4b0e4ec 100644
--- a/packages/palette_generator/test/palette_generator_test.dart
+++ b/packages/palette_generator/test/palette_generator_test.dart
@@ -28,10 +28,10 @@ class FakeImageProvider extends ImageProvider<FakeImageProvider> {
   }
 
   @override
-  // TODO(cyanglaz): migrate to use the new APIs
-  // https://github.com/flutter/flutter/issues/105336
-  // ignore: deprecated_member_use
-  ImageStreamCompleter load(FakeImageProvider key, DecoderCallback decode) {
+  ImageStreamCompleter loadImage(
+    FakeImageProvider key,
+    ImageDecoderCallback decode,
+  ) {
     assert(key == this);
     return OneFrameImageStreamCompleter(
       SynchronousFuture<ImageInfo>(