Change return code during unit tests