get unit tests working again, temporarily