Simple bug fix