Minor correction to get rid of compile warning