updated tests thankfully