Slight bug fix and testing note