[Extensions] Updating tests after load on demand landed

Two small fixes to our test suite.