Fixed broken import which causes a build failure