Remove unused argument
Add tests