couple of quick fixes