Fixed an import due to last commit