updated test and code for my edit bug fix