Fix failing test as it wasn't properly set