Tried to amend my previous commit to include this change, but did it wrong.