Fix broken import (sort of)