Fixed small var assignment error