Fixing up little errors