Fix failing test thanks @jmoody