Change conditional to check for ENV variable.