Fix for remove team tests