Modified infra script