Fixing mistake with missing source