Fixed an import that no longer exists.