Fix unit tests, had lingering local module