Fix another potantial bug.