Fixed the assignment error