A small fixes
Fix test