Fix #7, missing import