fixed edge issues