Fixed use of class that no longer exists.