Fix stupid assignment bug