Update tests to not use deprecated feature