Fixes broken rspec tests