diff --git a/cli/command/image/client_test.go b/cli/command/image/client_test.go
index 4e14bcf7abeb..869ab10676dc 100644
--- a/cli/command/image/client_test.go
+++ b/cli/command/image/client_test.go
@@ -26,7 +26,7 @@ type fakeClient struct {
 	imageListFunc    func(options image.ListOptions) ([]image.Summary, error)
 	imageInspectFunc func(img string) (image.InspectResponse, []byte, error)
 	imageImportFunc  func(source image.ImportSource, ref string, options image.ImportOptions) (io.ReadCloser, error)
-	imageHistoryFunc func(img string) ([]image.HistoryResponseItem, error)
+	imageHistoryFunc func(img string, options ...client.ImageHistoryOptions) ([]image.HistoryResponseItem, error)
 	imageBuildFunc   func(context.Context, io.Reader, types.ImageBuildOptions) (types.ImageBuildResponse, error)
 }