Minor bug
Finished fixing bug