Fixes copy/paste issue from last commit