Fix use statements. Remove erronous code assumption.