- Fixed the build error.  Had the wrong class to use before