minor bug fix, lost reference