Fix compile error (missing imports)