Fixed the failing test.