Fixed issue with last commit. (derp.)