fix issue with failing tests