Fixing unit tests post rebase Some changes that merged cleanly actually broke the unit tests, so this fixes them.