Updates unit tests