Oops, fixed missing import in last commit.