Fixed an oversight in previous update