Update short code test flow