fix broken unitttest, sry