Fixed problem with direction returned