Fix JNI test case after moving JNI impl to package