Fixed broken build in previous commit