Should have added this. Need to import the api module here for the REST to work.