Fix errors in build
Missing Class