Just correcting what I did wrong