Small change on previous patch.

Please ignore previous patch and use this one.