[dj] Fixing the broken test.