Fix little issue