Still trying to fix the exception