Add forgotten import to api module