Add int checking for axis