Workaround to avoid TF Lookup error