Updated ref to use lastest code from PR; not merge

Previously, a version of the code was used that was how a merge could look like.
Now, the code is used as it is in the PR