Fixes the failing test.