Fixed broken import that was missing from previous commit.