Fixed minor variable issue

Fix #234