Fix transform bug