Small lay-out fix