Small silly mistake in tests. Everything passes now.