lab5: fix