Slight modification

here is the patch: