Alter missed code from 374203fcfd2dab947c37d23358

Alter missed code from 374203fcfd2dab947c37d23358656b3f73a0f187