Move import check after build step