Fixed an exception I accidentally removed