Actually fixed Waila integration again.