Add missing @Test