Added a bug fix.