Fixed another potential problem