Changed import to local fork, while waiting for pull request to be
approved.