diff --git a/internal/file/file_service_operator.go b/internal/file/file_service_operator.go
index ddcff399a..cad5d68a4 100644
--- a/internal/file/file_service_operator.go
+++ b/internal/file/file_service_operator.go
@@ -542,6 +542,7 @@ func (fso *FileServiceOperator) sendFileUpdateStreamChunks(
 	// Ensure the stream is closed and wait for the server's response only
 	// after all chunks are sent
 	_, err = updateFileStreamClient.CloseAndRecv()
+
 	return err
 }
 
diff --git a/test/mock/grpc/mock_management_file_service.go b/test/mock/grpc/mock_management_file_service.go
index a865777c7..918b80672 100644
--- a/test/mock/grpc/mock_management_file_service.go
+++ b/test/mock/grpc/mock_management_file_service.go
@@ -217,6 +217,8 @@ func (mgs *FileService) UpdateFileStream(streamingServer grpc.ClientStreamingSer
 		return writeChunkedFileError
 	}
 
+	streamingServer.SendAndClose(&v1.UpdateFileResponse{})
+
 	return nil
 }