Fix build error for #2