Fix build with Omni