Adds width and height back to schema