Fixed broken build