Remove unused axis var

Axis is an unused var.