FIX BROKEN UNIT TEST