Fix some tests I missed