bug fix

See issue