Fix use statements