Fix build break in JNI layer