Fix attribute error from issue: #62