repaired unitTest