found out this is not the right way of doing it, and there is a much
better(tested) way, namely specify_shape