Add new comment

Al's picture

Thanks kasz, I've updated the code with this fix.