Minor change of remove unused use statement