Add the ability to set the tile entity