forgot to update the test file
Reorganized Tests