Fix silly mistake in test