A minor cleanup fix