Add a test for custom dimension