Move class from src/test to src/main (added by mistake).