add test
Small change