fix to tests