Fixed undefined variable (#5)