Fixed an import that caused a build error.