Tests and fixes for extension use

Extractor with just a function appears to have... issues though