Add tests for handling with axises