Fix line 128 and 129