quick fix to tests