Corrects assignment.
Merge pull request #373 from LearningLocker/issue/366

Corrects assignment.