Add test
Bug fix