Make extension tests a bit more robust.