Add test for PIT-2