Bug Fix v1.1

fixed the bug for version 1.1