BUG: Hack to get axis argument working.