Change the reshape to (1, 3) to be less permissive.