quick fix to get testing working again