Fix small assignment vs. equality bug