unit test based corrections