Update to make build passing on Travis