Update unitTest