I messed up the last commit... Here's a fix