fixed broken test suite