fixed some references
not sure what has changed here??