Added WAILA compatibility