Amend code to point to our fork