This works for imported targets as well.

Also update the test to call `UpdatePiece` as intended (now that the
test actually runs) and use `range` instead of `xrange`.