TODO Add the offset handling back