Tests: fixed missing extension register calls