Fixed another bug