fixed missing import in last commit, thanks Tim