Fixed my last PR

Sorry