Fix broken import

Import path was incorrect.