BUG: Fix loading model would not show color properly

We check the ""color by" mode too early so that the block colors are not updated properly.

Close #154 (closed)