Revert "Revert "minor fix""

This reverts commit 037a788f3b5b01cf312bd217036704d1e8e75c61.