Fix bug with extent shape