Fixed bug in umbrella wall potential