Fix #9 missing import