Fixing big mistake