Fix add_disk script