The exception to the 10% early withdrawal penalty for a first time home purchase is only if the funds were distributed from an IRA, not a 401(k).
IRS website for IRA exceptions - https://www.irs.gov/taxtopics/tc557
IRS website for non IRA exceptions - https://www.irs.gov/taxtopics/tc558
