Fix introduced return object error  (#157)