Adding a small piece to an existing test, great!