- fix the failing test