tbm_module: change if-statement into switch-case