Bug fix

Minor bug fixed