Update

Minor bug fix.