Fixing minor include bug