fix small stuff