Fix JNI build