Fix another bug :)