Add a test
fix bug