Remove unit test and add Rspec