You need to pass only `axis`