added a case for reshaping