fix reference error - Fixed #106 Fixed #105