Fixed bug 1

Fixed bug 1