this should fix our unit tests