small fix to tests