Make tests match readme