Fix rspec error

Finishes #1