fixed axis bug in EGKL2