fix some rough edges