Fixing extension tests