Bug Fix

Fix bug (re-tokenising)