tests: Update kleisli tests