Fixed an oversight in previous commit