Fix test suite
Bug fixes