[#547] Fix some extension tests