fix: update main script