Fix upgrade extension detection