another fix to tests