Make it possible to have the argument class different from the main
class.