fixed reshaping problem