Fixes broken import.