Moved the test to the test folder