fixed so rspec tests all pass