[BUGFIX] Add missing label

Partially fixes #453. PR is made against ``master`` due to the changed codebase.