Fixing the state init for key lengths 128 and 256.