Also fix unit test.
Also fix unit test.

Stanford-Online/edx-analytics-pipeline,Stanford-Online/edx-analytics-pipeline,edx/edx-analytics-pipeline,Stanford-Online/edx-analytics-pipeline,edx/edx-analytics-pipeline,edx/edx-analytics-pipeline