Fix: Had a class in the test package that needed to be in main