Fix build from mis-merge