Fixes

A couple small fixes.