Fix remaining warning