fix minor issue in assignment