Build fix for Windows

Missing imports when building on Windows