minor fixes
quick fix