Fix extension to use the correct variable for the name