shape as a tuple broke some tests and the function, now fixed