Add test for default case