Fixed wrong class name in the setup class