Fix error handling :S