Fixes tiny bugs