Add test
Fix test