Removes unused import and fixes small semantic error