Bug fix -(#231)