Quick bug fix.