Fixed last commit breaking the build