fixed intruduced bug