Minor test bugfix