Move back tests to the package