Fixed feature dim bug