Fix grunt dist task