Tried fixing test fail