Fixing little mistake