Missed generic parameter
Fixes for gradle build