Clean up unused import
Fix wrong arg name