Add case in sample class
Fix test