I see what you mean.

I want the lava to be at the same level as the floor so I don't want to put the edge lines you have between the two. As for the walls, I can see that might help a bit. Is the issue your having that you don't feel that the lava on the vace floor is lower than the central pillar?