Fix messed up test