Fix missing import & compile