Finish unit tests