Make them tests work now