Fix import for build