Bug fix (#1)