Updated test to not fail efter change.