fixing unit tests - thank you travis!