Update jest tests