Fix a last minut bug