Remove unnecessary method and add another test
Add another test