Fix small problem with comparison not being correct