Fixing transform bug