Fixed broken import due to the module name change