Little fix (not tested)