Tweak to compile