Fix the code written while way too tired.