Minor lazy bug fix