Change reference to api module in tests