Update specs to expect an element