move class from src/main to src/test, because we can