Add toggle to tests