Switched to Rspec for testing