Fixed warning caused by missing generic type