Minor bug fix (I know this is a bad commit, but...)