Bug fix.
Whole file will be changed anyways. So just made it work for the moment.