Fix unit test.
Merge pull request #3340 from edx/anton/fix-unit-test

Fix unit test.