small fix lab4