fixed typo and removed unneeded class