[RED] Fix single dimension test