Remove unused axis var

related to previous change, axis is not used anymore