Forgot the error handling.

My bad :(