Updated tests to use totem