Rake tests pass after updating unit tests