UPDATED FIX