Use reify instead of proxy
Handle reshape