Fixed assignment

Gods I made it harder than necessary...