Fixed broken import from moved class