Fix for small (2, 3) input