fix failed unit tests