Simple fixes

Two simple bug fixes.