Error message correction

Just some minor fixes