Fix code for error. Should run now.