- if (format != null) {
- checkArgument(format.equals(source.format()), "source’s format must equal this sink’s format");
- } else {
- format = source.format();
+ if ((connection != null) && (connection.source != null)) {
+ checkArgument(connection.source.format().equals(source.format()), "source’s format must equal this sink’s format");