Corrected shit so you can compile code

Do not commit if your code does not compile!