Correcting the way we pass a class