Fixing little errors