Moved tests to the correct class