Update test to match readme