Got it now! Sorry was having a hard time following the diff on Github