Update version script for the new API