Update. Fixes #0.5
layout change