Al final solo me había olvidado de multiplicar por i. Dejo mi resolución por si le sirve a alguien. Igual ya vi que esta resuelto en YouTube de otra manera.