Correct the root dir for class generation