Merge pull request #3224 from Ethanlm/STORM-3598

[STORM-3598] Fix UI visualization NullPointException