Fixed up issue with reference