Fixing the rspec unit tests