Updated the generator to set a type in rspec