Minor bug fixed

Minor bug fixed