Fix broken testing