Update tests files
Update unit tests