Fix extension problem