Updated config flow to conditionally call dis on test methods