Minor bugfix after renaming