A small modify

A small modify for testing