Fixed an bug. 1 to go