#337 bug fix.