finally works with 2 colours. test fail tests