Slight bug fix.