a little fix of previous code