Fix little bug