add extra case to test