fixed test failure hopefully