Fixes terrible code from last commit.