Fixes issue from my previous commit.