Updating unittests