correct errors

line 4 and 21