tests: fixed module import