Bug fix (#6)