Trying to fix the failing test.