fixed errors, wrong axis