Fix a few imports for testing an old func

These old functions weren't successfully importing into the tests.