Extension tests working as well