Fixed another little bug