Fix single test broken by the ref cleanup (#443)

Fix single test broken by the ref cleanup (#443) #2281