diff --git a/cli/command/image/client_test.go b/cli/command/image/client_test.go
index eb3a4cebef..8789a34789 100644
--- a/cli/command/image/client_test.go
+++ b/cli/command/image/client_test.go
@@ -69,7 +69,7 @@ func (cli *fakeClient) Info(_ context.Context) (system.Info, error) {
 
 func (cli *fakeClient) ImagePull(_ context.Context, ref string, options image.PullOptions) (io.ReadCloser, error) {
 	if cli.imagePullFunc != nil {
-		cli.imagePullFunc(ref, options)
+		return cli.imagePullFunc(ref, options)
 	}
 	return io.NopCloser(strings.NewReader("")), nil
 }