Compilation fix to prev commit.