Just a small fix to make stuff work again after the pull request. This went rather smoothly!