fix: use file extension to avoid import errors for vitest and jest wrappers

fix: use file extension to avoid import errors for vitest and jest wrappers #243