Fix location tests