Fixed an assignment error