Fix assignment3 compile issue