Fix little(but important bug)