Add null check for listener