fix import to match latest changes in service