@cdevroe think I did it! Was my first pull request, so not sure I did it right... looks like it worked though.