Bugfixes

@mridgway @redonkulus @lingyan