Add missing test cases