fixed bug, done with the tut