Fixed the dimension bug for real

With some weird shit - made no sense - whatever.