couple fixes

couple fixes