Fix another bug