Fixed snafu from last commit