Fix call to the class in the main program