fix small missing pieces