Update unittest