Quick bug fix