Fixing one more compilation problem