Make test to work
Fix lint error