Fix grunt build task