small fix to reshape function