Fix #4
Fix unused import