Add missing @Test annotation