Update unit tests to match previous change (derp)