minor fix to stuff