i think this is not needed (axis='sample')