Fix use statements for last class