Update unit tests I should have for commit 0f3e4ffe