minor fixes

@erikzaadi please review