tried to fix references
Hope it worked