fix to tests file