Add missing require - need rspec/expectations for this to work