Fixed import error with unittests

Needed to save old functions before overwriting them with lambdas.