Betterish testing

Just adding some missing things.