Rename context to dimension
Fix Dimension in integration tests