Source fix

minor fixes