Modify density keys naming
Update test to match new density naming