bug fix #3

Resolved #3