Add missing dependency to fix the build