Add a missing test