Fix grunt task