can't u just do `1 - ax`?