Update unit tests (temporary)