Fixed error if an error occurs :)