Clean up imports, remove unused variable
Missed a file