Corrected improper class and tests are green again =3
Removed .only from tests