Add missing @Test annotations to a few methods