Add and fix another test