Fix missing type parameter that broke the build