fixed a bug I made from the previous revision 

my initial patch wasn't totally backwards compatible, with this extra tweak it will be!