Quick little bug fix.